flake8

Lightning-AI · May 28, 2020 · b83aaa4 · b83aaa4
1 parent c6bffa8
commit b83aaa4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pytorch_lightning/metrics/converters.py b/pytorch_lightning/metrics/converters.py
@@ -196,7 +196,7 @@ def numpy_metric(group: Optional[Any] = None,
     """
 
     def decorator_fn(func_to_decorate):
-        return _apply_to_outputs(apply_to_collection, torch.Tensor, _sync_ddp,
+        return _apply_to_outputs(apply_to_collection, torch.Tensor, _sync_ddp_if_available,
                                  group=group,
                                  reduce_op=reduce_op)(_numpy_metric_conversion(func_to_decorate))
 
@@ -221,9 +221,8 @@ def tensor_metric(group: Optional[Any] = None,
 
     """
 
-
     def decorator_fn(func_to_decorate):
-        return _apply_to_outputs(apply_to_collection, torch.Tensor, _sync_ddp,
+        return _apply_to_outputs(apply_to_collection, torch.Tensor, _sync_ddp_if_available,
                                  group=group,
                                  reduce_op=reduce_op)(_tensor_metric_conversion(func_to_decorate))