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 f5449722..fc402f2c 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 @@ -155,7 +155,13 @@ public void setSha256(Object s) { * {@inheritDoc} */ public String getSource() { - return source; + if (source != null && + source.startsWith(Constants.ZENODO_DOMAIN) + && source.endsWith(Constants.ZENODO_ANNOYING_SUFFIX)) + return source.substring(0, + source.length() - Constants.ZENODO_ANNOYING_SUFFIX.length()); + else + return source; } /** 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 b1c44133..a0604384 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 @@ -178,7 +178,13 @@ public void setSha256(Object s) { * {@inheritDoc} */ public String getSource() { - return source; + if (source != null && + source.startsWith(Constants.ZENODO_DOMAIN) + && source.endsWith(Constants.ZENODO_ANNOYING_SUFFIX)) + return source.substring(0, + source.length() - Constants.ZENODO_ANNOYING_SUFFIX.length()); + else + return source; } /** 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 52d98871..076226ca 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 @@ -155,7 +155,13 @@ public void setSha256(Object s) { * {@inheritDoc} */ public String getSource() { - return source; + if (source != null && + source.startsWith(Constants.ZENODO_DOMAIN) + && source.endsWith(Constants.ZENODO_ANNOYING_SUFFIX)) + return source.substring(0, + source.length() - Constants.ZENODO_ANNOYING_SUFFIX.length()); + else + return source; } /** 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 c9ce9505..a6966051 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 @@ -25,6 +25,7 @@ import java.util.Map; import java.util.Set; +import io.bioimage.modelrunner.utils.Constants; import io.bioimage.modelrunner.versionmanagement.SupportedVersions; /** @@ -173,7 +174,13 @@ public void setSha256(Object s) { * {@inheritDoc} */ public String getSource() { - return source; + if (source != null && + source.startsWith(Constants.ZENODO_DOMAIN) + && source.endsWith(Constants.ZENODO_ANNOYING_SUFFIX)) + return source.substring(0, + source.length() - Constants.ZENODO_ANNOYING_SUFFIX.length()); + else + return source; } /** 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 9c553a3c..83cb24c6 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 @@ -180,7 +180,13 @@ public void setSha256(Object s) { * {@inheritDoc} */ public String getSource() { - return source; + if (source != null && + source.startsWith(Constants.ZENODO_DOMAIN) + && source.endsWith(Constants.ZENODO_ANNOYING_SUFFIX)) + return source.substring(0, + source.length() - Constants.ZENODO_ANNOYING_SUFFIX.length()); + else + return source; } /**