Upgrade MONAI (#10)

* upgrade monai to 1.2.0 remove previous warning suppressor * remove unused import
mehta-lab · Jun 12, 2023 · 69ec0a9 · 69ec0a9
1 parent 9fd175e
commit 69ec0a9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 13 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ dependencies = [
     "torchvision>=0.15.1",
     "tensorboard>=2.13.0",
     "lightning>=2.0.1",
-    "monai>=1.1.0",
+    "monai>=1.2.0",
     "jsonargparse[signatures]>=4.20.1",
     "scikit-image>=0.19.2",
     "matplotlib",

diff --git a/viscy/cli/cli.py b/viscy/cli/cli.py
@@ -1,4 +1,3 @@
-import warnings
 from datetime import datetime
 
 import torch
@@ -49,14 +48,8 @@ def add_arguments_to_parser(self, parser):
 
 def main():
     torch.set_float32_matmul_precision("high")
-    # TODO: remove this after MONAI 1.2 release
-    # https://github.com/Project-MONAI/MONAI/pull/6105
-    with warnings.catch_warnings():
-        warnings.filterwarnings(
-            "ignore", category=UserWarning, message="TypedStorage"
-        )
-        _ = VSLightningCLI(
-            model_class=PhaseToNuc25D,
-            datamodule_class=HCSDataModule,
-            trainer_class=VSTrainer,
-        )
+    _ = VSLightningCLI(
+        model_class=PhaseToNuc25D,
+        datamodule_class=HCSDataModule,
+        trainer_class=VSTrainer,
+    )