removenan keys for prediction

AllenCellModeling · May 2, 2024 · 72af355 · 72af355
1 parent 251f691
commit 72af355
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/configs/data/im2im/segmentation_plugin.yaml b/configs/data/im2im/segmentation_plugin.yaml
@@ -151,6 +151,7 @@ transforms:
   predict:
     _target_: monai.transforms.Compose
     transforms:
+      - _target_: cyto_dl.datamodules.dataframe.utils.RemoveNaNKeysd
       # load
       - _target_: monai.transforms.LoadImaged
         keys: ${source_col}