From d52ec739ba90658d2e5efb1ce92f2220f2260864 Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Tue, 19 Nov 2024 12:55:16 +0100
Subject: [PATCH] correction for when unzipping is needed

---
 .../modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java
index 0b81e25..d71b636 100644
--- a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java
+++ b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java
@@ -70,6 +70,7 @@
 import org.tensorflow.proto.framework.MetaGraphDef;
 import org.tensorflow.proto.framework.SignatureDef;
 import org.tensorflow.proto.framework.TensorInfo;
+import org.yaml.snakeyaml.Yaml;
 
 /**
  * Class to that communicates with the dl-model runner, see 
@@ -653,6 +654,7 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
         String gsonPath = getPathFromClass(Gson.class);
         String jnaPath = getPathFromClass(com.sun.jna.Library.class);
         String jnaPlatformPath = getPathFromClass(com.sun.jna.platform.FileUtils.class);
+        String snakeYaml = getPathFromClass(Yaml.class);
         if (modelrunnerPath == null || (modelrunnerPath.endsWith("DeepLearningEngineInterface.class") 
         		&& !modelrunnerPath.contains(File.pathSeparator)))
         	modelrunnerPath = System.getProperty("java.class.path");
@@ -660,6 +662,7 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
         classpath =  classpath + gsonPath + File.pathSeparator;
         classpath =  classpath + jnaPath + File.pathSeparator;
         classpath =  classpath + jnaPlatformPath + File.pathSeparator;
+        classpath =  classpath + snakeYaml + File.pathSeparator;
 
         return classpath;
     }