CM-2344: Implemented example of adding stage to the registered model's version.

Author: yaricom

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

@@ -44,6 +44,7 @@ public class RegistryModelExample {
     static final String SOME_MODEL_VERSION = "1.0.0";
     static final String SOME_MODEL_VERSION_UP = "1.0.1";
     static final String STAGE_PRODUCTION = "production";
+    static final String STAGE_STAGING = "staging";
     static final String SOME_NOTES = "some model notes";
 
     /**
@@ -147,6 +148,13 @@ record = api.registerModel(updatedModel, experiment.getExperimentKey());
                 System.out.printf("Overview of the model '%s' not found\n", registryName);
             }
 
+            // add stage to the model
+            //
+            System.out.printf("Adding stage `%s' to the registered model version '%s:%s'\n",
+                    STAGE_STAGING, registryName, SOME_MODEL_VERSION_UP);
+            api.addRegistryModelVersionStage(registryName, experiment.getWorkspaceName(),
+                    SOME_MODEL_VERSION_UP, STAGE_STAGING);
+
             // get details about model version
             //
             System.out.printf("Retrieving details of the model version '%s:%s'\n", registryName, SOME_MODEL_VERSION_UP);