File tree Expand file tree Collapse file tree
src/main/java/io/bioimage/modelrunner/tensorflow/v2/api050 Expand file tree Collapse file tree Original file line number Diff line number Diff line change 7272import org .tensorflow .proto .framework .SignatureDef ;
7373import org .tensorflow .proto .framework .TensorInfo ;
7474import 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 }
You can’t perform that action at this time.
0 commit comments