autofix

Signed-off-by: R. Garcia-Dias <[email protected]>

Author: garciadias

monai/utils/jupyter_utils.py

@@ -234,7 +234,7 @@ def plot_engine_status(
 
 
 def _get_loss_from_output(
-    output: list[torch.Tensor | dict[str, torch.Tensor]] | dict[str, torch.Tensor] | torch.Tensor,
+    output: list[torch.Tensor | dict[str, torch.Tensor]] | dict[str, torch.Tensor] | torch.Tensor
 ) -> torch.Tensor:
     """Returns a single value from the network output, which is a dict or tensor."""
 

tests/transforms/test_gibbs_noise.py

@@ -13,6 +13,7 @@
 
 import unittest
 from copy import deepcopy
+from itertools import product
 
 import numpy as np
 from parameterized import parameterized
@@ -21,7 +22,7 @@
 from monai.transforms import GibbsNoise
 from monai.utils.misc import set_determinism
 from monai.utils.module import optional_import
-from tests.test_utils import TEST_NDARRAYS, assert_allclose, dict_product
+from tests.test_utils import TEST_NDARRAYS, assert_allclose
 
 _, has_torch_fft = optional_import("torch.fft", name="fftshift")