chore: rename stage

ICIJ · Nov 13, 2024 · d85f65e · d85f65e
1 parent 19dc71a
commit d85f65e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/datashare-api/src/main/java/org/icij/datashare/Stage.java b/datashare-api/src/main/java/org/icij/datashare/Stage.java
@@ -11,7 +11,7 @@ public enum Stage {
     INDEX(true),
     ENQUEUEIDX(false),
     NLP(true),
-    CREATENLPBATCHESFROMINDEX(false),
+    CREATENLPBATCHESFROMIDX(false),
     BATCHNLP(false),
     ARTIFACT(false);
 

diff --git a/datashare-api/src/test/java/org/icij/datashare/PipelineHelperTest.java b/datashare-api/src/test/java/org/icij/datashare/PipelineHelperTest.java
@@ -52,7 +52,7 @@ public void test_get_queue_names_for_batch_nlp_pipeline() {
     @Test
     public void test_get_queue_names_for_batch_nlp_pipeline_from_index() {
         PipelineHelper pipelineHelper = new PipelineHelper(new PropertiesProvider(new HashMap<>() {{
-            put("stages", "CREATENLPBATCHESFROMINDEX,BATCHNLP");
+            put("stages", "CREATENLPBATCHESFROMIDX,BATCHNLP");
         }}));
         assertThat(pipelineHelper.getQueueNameFor(Stage.BATCHNLP)).isEqualTo("extract:queue:batchnlp");
     }

diff --git a/datashare-app/src/main/java/org/icij/datashare/CliApp.java b/datashare-app/src/main/java/org/icij/datashare/CliApp.java
@@ -22,7 +22,6 @@
 import org.slf4j.LoggerFactory;
 
 import java.io.IOException;
-import java.nio.file.Path;
 import java.util.List;
 import java.util.Properties;
 
@@ -137,7 +136,7 @@ private static void runTaskWorker(CommonMode mode, Properties properties) throws
                     new Task<>(EnqueueFromIndexTask.class.getName(), nullUser(), propertiesToMap(properties)));
         }
 
-        if (pipeline.has(Stage.CREATENLPBATCHESFROMINDEX)) {
+        if (pipeline.has(Stage.CREATENLPBATCHESFROMIDX)) {
             taskFactory.createBatchEnqueueFromIndexTask(
                     new Task<>(CreateNlpBatchesFromIndex.class.getName(), nullUser(), propertiesToMap(properties)),
                     (percentage) -> {logger.info("percentage: {}% done", percentage); return null;}).call();