From 65c6d4f65ebdf99bd1991eb44e2c1d1fadc72826 Mon Sep 17 00:00:00 2001
From: k-dominik <k-dominik@users.noreply.github.com>
Date: Mon, 25 Sep 2023 18:03:06 +0200
Subject: [PATCH] explicit export source for object classification

in the past nothing was passed to the workflow, which defaulted to
`Object Predictions` output.
This makes it explicit and fixes the previous binding to input type.
While it would be cool to choose the export source, and we should
definitely do it, see #64.
---
 .../ilastik4ij/workflow/ObjectClassificationCommand.java  | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java b/src/main/java/org/ilastik/ilastik4ij/workflow/ObjectClassificationCommand.java
index 27023c6..18034c9 100644
--- 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