skip pytorch2.6 test in auto3d

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

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/integration/test_auto3dseg_ensemble.py

@@ -33,7 +33,7 @@
 from monai.utils import check_parent_dir, optional_import, set_determinism
 from monai.utils.enums import AlgoKeys
 from tests.test_utils import (
-    SkipIfBeforePyTorchVersion,
+    SkipIfAtLeastPyTorchVersion,
     get_testing_algo_template_path,
     skip_if_downloading_fails,
     skip_if_no_cuda,
@@ -109,7 +109,7 @@ def create_sim_data(dataroot, sim_datalist, sim_dim, **kwargs):
 
 @skip_if_quick
 @skip_if_no_cuda
-@SkipIfBeforePyTorchVersion((1, 11, 1))
+@SkipIfAtLeastPyTorchVersion((2, 6, 0))
 @unittest.skipIf(not has_tb, "no tensorboard summary writer")
 class TestEnsembleBuilder(unittest.TestCase):
     def setUp(self) -> None: