Fix optional import

phisanti · phisanti · commit ce158865f10f · 2025-02-07T15:44:11.000+01:00
diff --git a/tests/test_CABlock.py b/tests/test_CABlock.py
@@ -21,7 +21,7 @@
 from monai.utils import optional_import
 from tests.utils import SkipIfBeforePyTorchVersion, assert_allclose
 
-einops, has_einops = optional_import("einops")
+rearrange, _ = optional_import("einops", name="rearrange")
 
 
 TEST_CASES_CAB = []
