Skip to content

Commit 2fccea4

Browse files
committed
Check if TODO persists
1 parent 7be120c commit 2fccea4

File tree

1 file changed

+1
-2
lines changed
  • pytorch_lightning/trainer/connectors/logger_connector

1 file changed

+1
-2
lines changed

pytorch_lightning/trainer/connectors/logger_connector/result.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,7 @@
2828
from pytorch_lightning.utilities.metrics import metrics_to_scalars
2929

3030
# re-define the ones from pytorch_lightning.utilities.types without the `Number` type
31-
# TODO(@tchaton): Typing-pickle issue on python<3.7 (https://github.com/cloudpipe/cloudpickle/pull/318)
32-
_METRIC = Any # Union[Metric, torch.Tensor]
31+
_METRIC = Union[Metric, torch.Tensor]
3332
_METRIC_COLLECTION = Union[_METRIC, Mapping[str, _METRIC]]
3433

3534

0 commit comments

Comments
 (0)