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 index 648827d430..4f2c644963 100644 --- 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 index a07a525a41..d5d2236650 100644 --- 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)