From faa943d9e0ba6b692a2680d88b665b895a44640b Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 30 Oct 2023 11:55:11 +0100
Subject: [PATCH] No longer need cleanup script, getting correct results

---
 .../transfer_labels_to_ilastik_object_classifier.py         | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py b/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py
index 748212ce..d9696e5a 100644
--- a/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py
+++ b/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py
@@ -98,7 +98,7 @@ def generate_ilastik_object_classifier(
         # delete existing classification weights
         if (k := f'ObjectExtraction/RegionFeatures/{lane:04d}') in h5.keys():
             del h5[k]
-        if (k := 'ObjectClassification/ClassificationForests') in h5.keys():
+        if (k := 'ObjectClassification/ClassifierForests') in h5.keys():
             del h5[k]
 
     return new_ilp
@@ -156,9 +156,7 @@ if __name__ == '__main__':
         label_names,
     )
 
-    from extensions.chaeo.examples.test_modify_object_classifier import modify_ilastik_object_classifier
-    auto_ilp = modify_ilastik_object_classifier(new_ilp)
-    # auto_ilp = new_ilp
+    auto_ilp = new_ilp
 
     def infer_and_compare(ilp, prefix, raw, mask, labels):
         mod = PatchStackObjectClassifier({'project_file': root / ilp})
-- 
GitLab