From 93bd76b0fd23309335a11c013f654d4a348c95b8 Mon Sep 17 00:00:00 2001 From: thomas chaton Date: Thu, 24 Jun 2021 12:58:58 +0100 Subject: [PATCH] [Refactor] Remove additional condition (#311) --- torchmetrics/metric.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchmetrics/metric.py b/torchmetrics/metric.py index e250608bfd5..b26799ef0aa 100644 --- 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])