changed message for scalling request

pytorch · May 7, 2020 · 2acfb62 · 2acfb62
1 parent f4b3cc8
commit 2acfb62
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/frontend/server/src/main/java/org/pytorch/serve/http/ManagementRequestHandler.java b/frontend/server/src/main/java/org/pytorch/serve/http/ManagementRequestHandler.java
@@ -244,7 +244,7 @@ private void handleRegisterModel(
                             + modelName
                             + "\" Version: "
                             + archive.getModelVersion()
-                            + " registered with 0 initial workers. Use scale API to add workers for the model.";
+                            + " registered with 0 initial workers. Use scale workers API to add workers for the model.";
             SnapshotManager.getInstance().saveSnapshot();
             NettyUtils.sendJsonResponse(ctx, new StatusResponse(msg));
             return;
@@ -337,7 +337,8 @@ private void updateModelWorkers(
                                     modelManager.scaleRequestStatus(modelName, modelVersion);
                             if (HttpResponseStatus.OK.equals(v)) {
                                 if (status) {
-                                    String msg = "Workers scaled";
+                                    String msg =
+                                            minWorkers + " Workers scaled for model " + modelName;
                                     if (isModelRegistrationRequest) {
                                         msg =
                                                 "Model \""

diff --git a/frontend/server/src/test/java/org/pytorch/serve/ModelServerTest.java b/frontend/server/src/test/java/org/pytorch/serve/ModelServerTest.java
@@ -284,7 +284,7 @@ private void testLoadModel(Channel channel, String url, String modelName, String
                         + modelName
                         + "\" Version: "
                         + version
-                        + " registered with 0 initial workers. Use scale API to add workers for the model.");
+                        + " registered with 0 initial workers. Use scale workers API to add workers for the model.");
     }
 
     private void testLoadModelFromURL(Channel channel) throws InterruptedException {
@@ -360,7 +360,7 @@ private void testSyncScaleModel(Channel channel, String modelName, String versio
         TestUtils.getLatch().await();
 
         StatusResponse resp = JsonUtils.GSON.fromJson(TestUtils.getResult(), StatusResponse.class);
-        Assert.assertEquals(resp.getStatus(), "Workers scaled");
+        Assert.assertEquals(resp.getStatus(), "1 Workers scaled for model " + modelName);
     }
 
     private void testUnregisterModel(Channel channel, String modelName, String version)