Ignore warning farther up

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

Author: ericspod

monai/__init__.py

@@ -43,6 +43,8 @@ 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,22 +36,20 @@
 __all__ = ["DiNTS", "TopologyConstruction", "TopologyInstance", "TopologySearch"]
 
 
-# TODO: added temporarily for PyTorch 2.14 warnings, remove when factoring out deprecated Torchscript components
-with warnings.catch_warnings(action="ignore"):
+@torch.jit.interface
+class CellInterface(torch.nn.Module):
+    """interface for torchscriptable Cell"""
 
-    @torch.jit.interface
-    class CellInterface(torch.nn.Module):
-        """Abstract interface for Cell modules used in DiNTS."""
+    def forward(self, x: torch.Tensor, weight: torch.Tensor | None) -> torch.Tensor:  # type: ignore
+        pass
 
-        def forward(self, x: torch.Tensor, weight: torch.Tensor | None) -> torch.Tensor:  # type: ignore
-            pass
 
-    @torch.jit.interface
-    class StemInterface(torch.nn.Module):
-        """Abstract interface for Stem modules used in DiNTS."""
+@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):