k6 fixes (#4841)

SeldonIO · May 11, 2023 · 9de2bc6 · 9de2bc6
1 parent 37cedf8
commit 9de2bc6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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)
     }
 }