File tree 1 file changed +3
-0
lines changed
src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050
1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change 72
72
import org .tensorflow .proto .framework .SignatureDef ;
73
73
import org .tensorflow .proto .framework .TensorInfo ;
74
74
import org .tensorflow .types .family .TType ;
75
+ import org .yaml .snakeyaml .Yaml ;
75
76
76
77
/**
77
78
* Class to that communicates with the dl-model runner, see
@@ -617,13 +618,15 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
617
618
String gsonPath = getPathFromClass (Gson .class );
618
619
String jnaPath = getPathFromClass (com .sun .jna .Library .class );
619
620
String jnaPlatformPath = getPathFromClass (com .sun .jna .platform .FileUtils .class );
621
+ String snakeYaml = getPathFromClass (Yaml .class );
620
622
if (modelrunnerPath == null || (modelrunnerPath .endsWith ("DeepLearningEngineInterface.class" )
621
623
&& !modelrunnerPath .contains (File .pathSeparator )))
622
624
modelrunnerPath = System .getProperty ("java.class.path" );
623
625
String classpath = modelrunnerPath + File .pathSeparator + imglib2Path + File .pathSeparator ;
624
626
classpath = classpath + gsonPath + File .pathSeparator ;
625
627
classpath = classpath + jnaPath + File .pathSeparator ;
626
628
classpath = classpath + jnaPlatformPath + File .pathSeparator ;
629
+ classpath = classpath + snakeYaml + File .pathSeparator ;
627
630
628
631
return classpath ;
629
632
}
You can’t perform that action at this time.
0 commit comments