diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/KerasWeights.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/KerasWeights.java
index f5c247e0..7682d54b 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/KerasWeights.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/KerasWeights.java
@@ -305,7 +305,7 @@ public boolean isSupportGPU() {
 	/**
 	 * {@inheritDoc}
 	 */
-	public String getJDLLCompatibleToTrainingVersion() {
+	public String getJavaTrainingVersion() {
 		return compatibleVersion;
 	}
 	
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/OnnxWeights.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/OnnxWeights.java
index 8fdc4862..58db60e5 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/OnnxWeights.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/OnnxWeights.java
@@ -311,7 +311,7 @@ public boolean isSupportGPU() {
 	/**
 	 * {@inheritDoc}
 	 */
-	public String getJDLLCompatibleToTrainingVersion() {
+	public String getJavaTrainingVersion() {
 		return compatibleVersion;
 	}
 	
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/PytorchWeights.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/PytorchWeights.java
index 9ad988a9..22bd1a5a 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/PytorchWeights.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/PytorchWeights.java
@@ -307,7 +307,7 @@ public boolean isSupportGPU() {
 	/**
 	 * {@inheritDoc}
 	 */
-	public String getJDLLCompatibleToTrainingVersion() {
+	public String getJavaTrainingVersion() {
 		return compatibleVersion;
 	}
 	
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfJsWeights.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfJsWeights.java
index 2977baf8..898902e2 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfJsWeights.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfJsWeights.java
@@ -309,7 +309,7 @@ public boolean isSupportGPU() {
 	/**
 	 * {@inheritDoc}
 	 */
-	public String getJDLLCompatibleToTrainingVersion() {
+	public String getJavaTrainingVersion() {
 		return compatibleVersion;
 	}
 	
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfWeights.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfWeights.java
index 171b8aab..ac3d43ee 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfWeights.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TfWeights.java
@@ -306,7 +306,7 @@ public boolean isSupportGPU() {
 	/**
 	 * {@inheritDoc}
 	 */
-	public String getJDLLCompatibleToTrainingVersion() {
+	public String getJavaTrainingVersion() {
 		return compatibleVersion;
 	}
 	
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TorchscriptWeights.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TorchscriptWeights.java
index 5c679ae3..01876110 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TorchscriptWeights.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/TorchscriptWeights.java
@@ -311,7 +311,7 @@ public boolean isSupportGPU() {
 	/**
 	 * {@inheritDoc}
 	 */
-	public String getJDLLCompatibleToTrainingVersion() {
+	public String getJavaTrainingVersion() {
 		return compatibleVersion;
 	}
 	
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/WeightFormat.java b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/WeightFormat.java
index 8c5095dc..834f894f 100644
--- a/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/WeightFormat.java
+++ b/src/main/java/io/bioimage/modelrunner/bioimageio/description/weights/WeightFormat.java
@@ -41,12 +41,12 @@ public interface WeightFormat {
 	public String getTrainingVersion();
 	
 	/**
-	 * Return the version supported by JDLL that it is compatible with
+	 * Return the Java version of the framework supported by JDLL that it is compatible with
 	 * the training version of the weights
-	 * @return the version supported by JDLL that it is compatible with
+	 * @return the Java version of the framework supported by JDLL that it is compatible with
 	 * the training version of the weights
 	 */
-	public String getJDLLCompatibleToTrainingVersion();
+	public String getJavaTrainingVersion();
 	
 	/**
 	 * Format of the weights of the model. The supported weights by the Bioimage.io are: