unify nickname and id

bioimage-io · Nov 12, 2024 · e07324a · e07324a
1 parent 7070a45
commit e07324a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/bioimageio/BioimageioRepo.java b/src/main/java/io/bioimage/modelrunner/bioimageio/BioimageioRepo.java
@@ -424,13 +424,14 @@ public ModelDescriptor selectByID(String modelID) {
 		Entry<String, ModelDescriptor> modelEntry = this.listAllModels(false).entrySet().stream()
 				.filter(ee -> {
 					String id = ee.getValue().getModelID();
+					String nickname = ee.getValue().getNickname();
 					if (id.length() - id.replace("/", "").length() == 2) {
 						id = id.substring(0, id.lastIndexOf("/"));
 					}
 					if (modelID.length() - modelID.replace("/", "").length() == 2) {
-						return modelID.substring(0, modelID.lastIndexOf("/")).equals(id);
+						return modelID.substring(0, modelID.lastIndexOf("/")).equals(id) || modelID.equals(nickname);
 					}
-					return modelID.equals(id);
+					return modelID.equals(id) || modelID.equals(nickname);
 				}).findFirst().orElse(null);
 		if (modelEntry != null)
 			return modelEntry.getValue();