diff --git a/tests/test_auto3dseg_ensemble.py b/tests/test_auto3dseg_ensemble.py index a1e1f87b4a..ad36b883d9 100644 --- a/tests/test_auto3dseg_ensemble.py +++ b/tests/test_auto3dseg_ensemble.py @@ -76,7 +76,7 @@ @skip_if_quick -@SkipIfBeforePyTorchVersion((1, 9, 1)) +@SkipIfBeforePyTorchVersion((1, 11, 1)) @unittest.skipIf(not has_tb, "no tensorboard summary writer") class TestEnsembleBuilder(unittest.TestCase): def setUp(self) -> None: diff --git a/tests/test_auto3dseg_hpo.py b/tests/test_auto3dseg_hpo.py index 3aac4afebd..f8b9ebb2fb 100644 --- a/tests/test_auto3dseg_hpo.py +++ b/tests/test_auto3dseg_hpo.py @@ -75,7 +75,7 @@ def skip_if_no_optuna(obj): } -@SkipIfBeforePyTorchVersion((1, 9, 1)) +@SkipIfBeforePyTorchVersion((1, 11, 1)) @unittest.skipIf(not has_tb, "no tensorboard summary writer") class TestHPO(unittest.TestCase): def setUp(self) -> None: