Fix again

Signed-off-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>

Author: ericspod

monai/__init__.py

@@ -43,8 +43,6 @@ def filter(self, record):
             "class-based `config`",
             "pkg_resources",
             "Implicitly cleaning up",
-            # TODO: added temporarily for PyTorch 2.14, remove when factoring out deprecated Torchscript components
-            "torch.jit.interface"
         ]
         for message in message_bodies_to_ignore:
             if message in record.getMessage():

monai/networks/nets/dints.py

@@ -36,20 +36,23 @@
 __all__ = ["DiNTS", "TopologyConstruction", "TopologyInstance", "TopologySearch"]
 
 
-@torch.jit.interface
-class CellInterface(torch.nn.Module):
-    """interface for torchscriptable Cell"""
+# TODO: added temporarily for PyTorch 2.14 warnings, remove when factoring out deprecated Torchscript components
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
 
-    def forward(self, x: torch.Tensor, weight: torch.Tensor | None) -> torch.Tensor:  # type: ignore
-        pass
+    @torch.jit.interface
+    class CellInterface(torch.nn.Module):
+        """interface for torchscriptable Cell"""
 
+        def forward(self, x: torch.Tensor, weight: torch.Tensor | None) -> torch.Tensor:  # type: ignore
+            pass
 
-@torch.jit.interface
-class StemInterface(torch.nn.Module):
-    """interface for torchscriptable Stem"""
+    @torch.jit.interface
+    class StemInterface(torch.nn.Module):
+        """interface for torchscriptable Stem"""
 
-    def forward(self, x: torch.Tensor) -> torch.Tensor:  # type: ignore
-        pass
+        def forward(self, x: torch.Tensor) -> torch.Tensor:  # type: ignore
+            pass
 
 
 class StemTS(StemInterface):