Reformat code with black

learning-at-home · Nov 4, 2024 · b462d79 · b462d79
1 parent 32ed45f
commit b462d79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hivemind/optim/grad_scaler.py b/hivemind/optim/grad_scaler.py
@@ -125,4 +125,4 @@ def _unscale_grads_(
 
     def are_grads_finite(self, optimizer: TorchOptimizer, use_cached: bool = False) -> bool:
         opt_dict = self._found_inf_per_device(optimizer) if use_cached else self._check_inf_per_device(optimizer)
-        return not sum(v.item() for v in opt_dict.values())
+        return not sum(v.item() for v in opt_dict.values())