Skip to content

Commit 1fc0a5b

Browse files
committed
reroll modified tests
Signed-off-by: R. Garcia-Dias <[email protected]>
1 parent 241e24c commit 1fc0a5b

File tree

4 files changed

+5
-4
lines changed

4 files changed

+5
-4
lines changed

MONAI

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit f97a0e9bc7787df261284db2cdeffa2e8d631512

monai/utils/jupyter_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ def plot_engine_status(
234234

235235

236236
def _get_loss_from_output(
237-
output: list[torch.Tensor | dict[str, torch.Tensor]] | dict[str, torch.Tensor] | torch.Tensor,
237+
output: list[torch.Tensor | dict[str, torch.Tensor]] | dict[str, torch.Tensor] | torch.Tensor
238238
) -> torch.Tensor:
239239
"""Returns a single value from the network output, which is a dict or tensor."""
240240

monai_src

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit f97a0e9bc7787df261284db2cdeffa2e8d631512

tests/transforms/test_gibbs_noise.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313

1414
import unittest
1515
from copy import deepcopy
16-
from itertools import product
1716

1817
import numpy as np
1918
from parameterized import parameterized
@@ -22,13 +21,13 @@
2221
from monai.transforms import GibbsNoise
2322
from monai.utils.misc import set_determinism
2423
from monai.utils.module import optional_import
25-
from tests.test_utils import TEST_NDARRAYS, assert_allclose
24+
from tests.test_utils import TEST_NDARRAYS, assert_allclose, dict_product
2625

2726
_, has_torch_fft = optional_import("torch.fft", name="fftshift")
2827

2928
shapes = ((128, 64), (64, 48, 80))
3029
input_types = TEST_NDARRAYS if has_torch_fft else [np.array]
31-
TEST_CASES = list(map(list, product(shape=shapes, input_type=input_types)))
30+
TEST_CASES = [[p_dict["shape"], p_dict["input_type"]] for p_dict in dict_product(shape=shapes, input_type=input_types)]
3231

3332

3433
class TestGibbsNoise(unittest.TestCase):

0 commit comments

Comments
 (0)