[Refactor] Remove additional condition (#311)

Lightning-AI · Jun 24, 2021 · 93bd76b · 93bd76b
1 parent c8f81b1
commit 93bd76b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmetrics/metric.py b/torchmetrics/metric.py
@@ -214,7 +214,7 @@ def _sync_dist(self, dist_sync_fn: Callable = gather_all_tensors, process_group:
 
         for attr, reduction_fn in self._reductions.items():
             # pre-processing ops (stack or flatten for inputs)
-            if isinstance(output_dict[attr], Sequence) and isinstance(output_dict[attr][0], Tensor):
+            if isinstance(output_dict[attr][0], Tensor):
                 output_dict[attr] = torch.stack(output_dict[attr])
             elif isinstance(output_dict[attr][0], list):
                 output_dict[attr] = _flatten(output_dict[attr])