From 99f05ed23f818d4f21c2c6925a66e75df606c859 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Sun, 20 Sep 2020 00:29:06 +0200 Subject: [PATCH] fix warnings on windows (#3555) --- pytorch_lightning/metrics/converters.py | 8 ++------ pytorch_lightning/metrics/sklearns.py | 8 ++------ 2 files changed, 4 insertions(+), 12 deletions(-) diff --git a/pytorch_lightning/metrics/converters.py b/pytorch_lightning/metrics/converters.py index b43462cddc..f62a470981 100644 --- a/pytorch_lightning/metrics/converters.py +++ b/pytorch_lightning/metrics/converters.py @@ -26,18 +26,14 @@ import numpy as np import torch from torch.utils.data._utils.collate import np_str_obj_array_pattern -from pytorch_lightning.utilities import rank_zero_warn from pytorch_lightning.utilities.apply_func import apply_to_collection -try: +if torch.distributed.is_available(): from torch.distributed import ReduceOp -except ImportError: - +else: class ReduceOp: SUM = None - rank_zero_warn("Unsupported `ReduceOp` for distributed computing") - def _apply_to_inputs(func_to_apply: Callable, *dec_args, **dec_kwargs) -> Callable: """ diff --git a/pytorch_lightning/metrics/sklearns.py b/pytorch_lightning/metrics/sklearns.py index 4cb7cca0a4..d19bd05e63 100644 --- a/pytorch_lightning/metrics/sklearns.py +++ b/pytorch_lightning/metrics/sklearns.py @@ -19,17 +19,13 @@ import torch from pytorch_lightning import _logger as lightning_logger from pytorch_lightning.metrics.metric import NumpyMetric -from pytorch_lightning.utilities import rank_zero_warn -try: +if torch.distributed.is_available(): from torch.distributed import group -except ImportError: - +else: class group: WORLD = None - rank_zero_warn("Unsupported `group` for distributed computing.") - class SklearnMetric(NumpyMetric): """