File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed
src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020 Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change 7070import org .tensorflow .proto .framework .MetaGraphDef ;
7171import org .tensorflow .proto .framework .SignatureDef ;
7272import 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 }
You can’t perform that action at this time.
0 commit comments