correction for when unzipping is needed

bioimage-io · Nov 19, 2024 · d52ec73 · d52ec73
1 parent cd45015
commit d52ec73
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java
@@ -70,6 +70,7 @@
 import org.tensorflow.proto.framework.MetaGraphDef;
 import org.tensorflow.proto.framework.SignatureDef;
 import org.tensorflow.proto.framework.TensorInfo;
+import org.yaml.snakeyaml.Yaml;
 
 /**
  * Class to that communicates with the dl-model runner, see 
@@ -653,13 +654,15 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
         String gsonPath = getPathFromClass(Gson.class);
         String jnaPath = getPathFromClass(com.sun.jna.Library.class);
         String jnaPlatformPath = getPathFromClass(com.sun.jna.platform.FileUtils.class);
+        String snakeYaml = getPathFromClass(Yaml.class);
         if (modelrunnerPath == null || (modelrunnerPath.endsWith("DeepLearningEngineInterface.class") 
         		&& !modelrunnerPath.contains(File.pathSeparator)))
         	modelrunnerPath = System.getProperty("java.class.path");
         String classpath =  modelrunnerPath + File.pathSeparator + imglib2Path + File.pathSeparator;
         classpath =  classpath + gsonPath + File.pathSeparator;
         classpath =  classpath + jnaPath + File.pathSeparator;
         classpath =  classpath + jnaPlatformPath + File.pathSeparator;
+        classpath =  classpath + snakeYaml + File.pathSeparator;
 
         return classpath;
     }