Merge pull request #119 from ilastik/fix-oc-command

explicit export source for object classification
ilastik · Sep 26, 2023 · 03934ae · 03934ae
2 parents 42d435f + 65c6d4f
commit 03934ae
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java b/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java
@@ -46,13 +46,7 @@ public final class ObjectClassificationCommand<T extends NativeType<T> & RealTyp
 
     @Override
     protected List<String> workflowArgs() {
-        if (ROLE_PROBABILITIES.equals(secondInputType)) {
-            return Collections.singletonList("--export_source=Object Probabilities");
-        }
-        if (ROLE_SEGMENTATION.equals(secondInputType)) {
-            return Collections.singletonList("--export_source=Object Predictions");
-        }
-        throw new IllegalStateException("Unexpected value: " + secondInputType);
+        return Collections.singletonList("--export_source=Object Predictions");
     }
 
     @Override