From 9de2bc681ce63a53b278f71f43914b95be8113c8 Mon Sep 17 00:00:00 2001 From: Sherif Akoush Date: Thu, 11 May 2023 13:33:49 +0100 Subject: [PATCH] k6 fixes (#4841) --- tests/k6/components/utils.js | 4 ++-- tests/k6/scenarios/infer_constant_rate.js | 2 +- tests/k6/scenarios/predict_agent_oc.js | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/k6/components/utils.js b/tests/k6/components/utils.js index c930b74eb7..25a2a4963b 100644 --- a/tests/k6/components/utils.js +++ b/tests/k6/components/utils.js @@ -44,7 +44,7 @@ export function setupBase(config ) { for (let j = 0; j < config.maxNumModels.length; j++) { const n = config.maxNumModels[j] - 1 const modelName = config.modelNamePrefix[j] + n.toString() - const modelNameWithVersion = modelName + getVersionSuffix(config) // first version + const modelNameWithVersion = modelName + getVersionSuffix(config.isSchedulerProxy) // first version while (modelStatusHttp(config.inferHttpEndpoint, config.isEnvoy?modelName:modelNameWithVersion, config.isEnvoy) !== 200) { sleep(1) } @@ -63,7 +63,7 @@ export function setupBase(config ) { for (let i = 0; i < config.maxNumModels[j]; i++) { const modelName = config.modelNamePrefix[j] + i.toString() - const modelNameWithVersion = modelName + getVersionSuffix(config) // first version + const modelNameWithVersion = modelName + getVersionSuffix(config.isSchedulerProxy) // first version const model = generateModel(config.modelType[j], modelNameWithVersion, 1, 1, config.isSchedulerProxy, config.modelMemoryBytes[j]) diff --git a/tests/k6/scenarios/infer_constant_rate.js b/tests/k6/scenarios/infer_constant_rate.js index 4d983bc3ab..b400913049 100644 --- a/tests/k6/scenarios/infer_constant_rate.js +++ b/tests/k6/scenarios/infer_constant_rate.js @@ -37,7 +37,7 @@ export default function (config) { const modelId = Math.floor(Math.random() * config.maxNumModels[idx]) const modelName = config.modelNamePrefix[idx] + modelId.toString() - const modelNameWithVersion = modelName + getVersionSuffix(config) // first version + const modelNameWithVersion = modelName + getVersionSuffix(config.isSchedulerProxy) // first version if (config.inferType === "REST") { doInfer(modelName, modelNameWithVersion, config, true, idx) diff --git a/tests/k6/scenarios/predict_agent_oc.js b/tests/k6/scenarios/predict_agent_oc.js index 28c74b8dbe..67e4a1cc4f 100644 --- a/tests/k6/scenarios/predict_agent_oc.js +++ b/tests/k6/scenarios/predict_agent_oc.js @@ -25,7 +25,7 @@ export default function (config) { for (let i = 0; i < config.maxNumModels[idx]; i++) { var modelName = config.modelNamePrefix[idx] + i.toString() - var modelNameWithVersion = modelName + getVersionSuffix(config) // first version + var modelNameWithVersion = modelName + getVersionSuffix(config.isSchedulerProxy) // first version doInfer(modelName, modelNameWithVersion, config, false, idx) } }