Skip to content

Commit 6d2f6bb

Browse files
committed
correction for when unzipping is needed
1 parent f7a6c2a commit 6d2f6bb

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050/Tensorflow2Interface.java

+3
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@
7272
import org.tensorflow.proto.framework.SignatureDef;
7373
import org.tensorflow.proto.framework.TensorInfo;
7474
import org.tensorflow.types.family.TType;
75+
import org.yaml.snakeyaml.Yaml;
7576

7677
/**
7778
* Class to that communicates with the dl-model runner, see
@@ -617,13 +618,15 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
617618
String gsonPath = getPathFromClass(Gson.class);
618619
String jnaPath = getPathFromClass(com.sun.jna.Library.class);
619620
String jnaPlatformPath = getPathFromClass(com.sun.jna.platform.FileUtils.class);
621+
String snakeYaml = getPathFromClass(Yaml.class);
620622
if (modelrunnerPath == null || (modelrunnerPath.endsWith("DeepLearningEngineInterface.class")
621623
&& !modelrunnerPath.contains(File.pathSeparator)))
622624
modelrunnerPath = System.getProperty("java.class.path");
623625
String classpath = modelrunnerPath + File.pathSeparator + imglib2Path + File.pathSeparator;
624626
classpath = classpath + gsonPath + File.pathSeparator;
625627
classpath = classpath + jnaPath + File.pathSeparator;
626628
classpath = classpath + jnaPlatformPath + File.pathSeparator;
629+
classpath = classpath + snakeYaml + File.pathSeparator;
627630

628631
return classpath;
629632
}

0 commit comments

Comments
 (0)