removenan keys for prediction (#380)

Co-authored-by: Benjamin Morris <[email protected]>
AllenCellModeling · May 2, 2024 · 14527ba · 14527ba
1 parent 251f691
commit 14527ba
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}