Skip to content

Commit b6e387e

Browse files
committed
Update SWAN configuration to detect all SRMs
1 parent 0eaa5ca commit b6e387e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

swan-assist/src/main/java/de/fraunhofer/iem/swan/assist/actions/RunSwanTask.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
import java.io.IOException;
2121
import java.time.LocalDateTime;
2222
import java.time.format.DateTimeFormatter;
23-
import java.util.Arrays;
2423
import java.util.HashMap;
24+
import java.util.List;
2525

2626
public class RunSwanTask extends Task.Backgroundable {
2727

@@ -45,7 +45,7 @@ public void run(@NotNull ProgressIndicator indicator) {
4545
options.setTestDataDir(parameters.get(Constants.SOURCE_DIRECTORY));
4646
options.setOutputDir(parameters.get(Constants.OUTPUT_DIRECTORY));
4747
options.setToolkit(parameters.get(Constants.TOOLKIT).toLowerCase());
48-
options.setSrmClasses(Arrays.asList("source", "sink"));
48+
options.setSrmClasses(List.of("all"));
4949
options.setPhase("predict");
5050
options.setTrainDataDir("");
5151

0 commit comments

Comments
 (0)