fix optiona import

phisanti · phisanti · commit f520e99c3a35 · 2025-02-07T14:13:59.000+01:00
diff --git a/monai/networks/blocks/cablock.py b/monai/networks/blocks/cablock.py
@@ -13,7 +13,9 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from einops import rearrange
+from monai.utils import optional_import
+
+rearrange, _ = optional_import("einops", name="rearrange")
 
 from monai.networks.blocks.convolutions import Convolution
 
