Skip to content

Commit d52ec73

Browse files
committed
correction for when unzipping is needed
1 parent cd45015 commit d52ec73

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Diff for: src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java

+3
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@
7070
import org.tensorflow.proto.framework.MetaGraphDef;
7171
import org.tensorflow.proto.framework.SignatureDef;
7272
import org.tensorflow.proto.framework.TensorInfo;
73+
import org.yaml.snakeyaml.Yaml;
7374

7475
/**
7576
* Class to that communicates with the dl-model runner, see
@@ -653,13 +654,15 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
653654
String gsonPath = getPathFromClass(Gson.class);
654655
String jnaPath = getPathFromClass(com.sun.jna.Library.class);
655656
String jnaPlatformPath = getPathFromClass(com.sun.jna.platform.FileUtils.class);
657+
String snakeYaml = getPathFromClass(Yaml.class);
656658
if (modelrunnerPath == null || (modelrunnerPath.endsWith("DeepLearningEngineInterface.class")
657659
&& !modelrunnerPath.contains(File.pathSeparator)))
658660
modelrunnerPath = System.getProperty("java.class.path");
659661
String classpath = modelrunnerPath + File.pathSeparator + imglib2Path + File.pathSeparator;
660662
classpath = classpath + gsonPath + File.pathSeparator;
661663
classpath = classpath + jnaPath + File.pathSeparator;
662664
classpath = classpath + jnaPlatformPath + File.pathSeparator;
665+
classpath = classpath + snakeYaml + File.pathSeparator;
663666

664667
return classpath;
665668
}

0 commit comments

Comments
 (0)