Skip to content

Commit bb1a56c

Browse files
committed
CM-2347: Implemented example how to get registered models count under specific workspace.
1 parent d1f5acb commit bb1a56c

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

comet-examples/src/main/java/ml/comet/examples/RegistryModelExample.java

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,17 @@ record = api.registerModel(updatedModel, experiment.getExperimentKey());
203203
newComment, newStages);
204204
System.out.printf("Model version details was successfully updated for: %s\n", SOME_MODEL_VERSION_UP);
205205

206+
// read the number of registered models
207+
//
208+
Optional<Integer> modelsCount = api.getRegistryModelsCount(experiment.getWorkspaceName());
209+
if (modelsCount.isPresent()) {
210+
System.out.printf("The number of registered models under workspace '%s' is: %d\n",
211+
experiment.getWorkspaceName(), modelsCount.get());
212+
} else {
213+
System.out.printf("Failed to get registered models count. The workspace '%s' doesn't exists.\n",
214+
experiment.getWorkspaceName());
215+
}
216+
206217
// delete registry model version
207218
//
208219
System.out.printf("Deleting registry model version '%s/%s:%s'.\n",

0 commit comments

Comments
 (0)