Fixing tests

KonduitAI · Apr 28, 2020 · 6019215 · 6019215
1 parent 1c59bdd
commit 6019215
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/konduit-serving-api/src/main/java/ai/konduit/serving/InferenceConfiguration.java b/konduit-serving-api/src/main/java/ai/konduit/serving/InferenceConfiguration.java
@@ -62,9 +62,11 @@ public void stepSetup(List<PipelineStep> steps) throws Exception {
  PythonStep pythonStep = (PythonStep) step;
  Map<String, PythonConfig> pythonConfigs = pythonStep.getPythonConfigs();
 
- for(String key : pythonConfigs.keySet()) {
- if(pythonStep.getInputNames() == null || !pythonStep.getInputNames().contains(key))
- pythonStep.step(key, pythonConfigs.get(key));
+ if (pythonConfigs != null) {
+ for (String key : pythonConfigs.keySet()) {
+ if (pythonStep.getInputNames() == null || !pythonStep.getInputNames().contains(key))
+ pythonStep.step(key, pythonConfigs.get(key));
+ }
  }
  }
 

diff --git a/...uit-serving-api/src/test/java/ai/konduit/serving/pipeline/PipelineSerializationTests.java b/...uit-serving-api/src/test/java/ai/konduit/serving/pipeline/PipelineSerializationTests.java
@@ -54,7 +54,5 @@ public void testPipelineSerialization() throws Exception {
  .step(imageLoadingStepConfig)
  .build();
  System.out.println(config.toJson());
-
  }
-
 }
diff --git a/python/konduit/load.py b/python/konduit/load.py
@@ -210,9 +210,11 @@ def get_python_step(step_config):
  :return: konduit.inference.PythonStep instance.
  """
 
- python_config = PythonConfig(**step_config)
- step = PythonStep().step(python_config)
- return step
+ python_step = PythonStep()
+ for key, python_config in step_config:
+ python_step.step(key, PythonConfig(**python_config))
+
+ return python_step
 
 
 def get_image_load_step(step_config):