From 72af355ade1a04f1587695d492c735b6db3e3151 Mon Sep 17 00:00:00 2001
From: Benjamin Morris
 <benjamin.morris@prd-aics-dcp-037.corp.alleninstitute.org>
Date: Thu, 2 May 2024 09:50:51 -0700
Subject: [PATCH] removenan keys for prediction

---
 configs/data/im2im/segmentation_plugin.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/configs/data/im2im/segmentation_plugin.yaml b/configs/data/im2im/segmentation_plugin.yaml
index f296e469..1ffa15e1 100644
--- 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}