Fix incorrect checks to remove pyfunc/docker ensembler cpu limits

caraml-dev · Jun 6, 2024 · 4cef9be · 4cef9be
1 parent cbebbd6
commit 4cef9be
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/ui/src/services/router/TuringRouter.js b/ui/src/services/router/TuringRouter.js
@@ -152,10 +152,16 @@ export class TuringRouter {
       if (obj.config.ensembler.type === "pyfunc") {
         // Delete the docker config
         delete obj.config["ensembler"].docker_config;
+        // Pyfunc ensembler CPU limit
+        if (obj.config["ensembler"].pyfunc_config.resource_request?.cpu_limit === "") {
+          delete obj.config["ensembler"].pyfunc_config.resource_request.cpu_limit;
+        }
       }
-      // Docker/Pyfunc ensembler CPU limit
-      if (obj.config.ensembler.resource_request?.cpu_limit === "") {
-        delete obj.config.ensembler.resource_request.cpu_limit;
+      if (obj.config.ensembler.type === "docker") {
+        // Docker ensembler CPU limit
+        if (obj.config["ensembler"].docker_config.resource_request?.cpu_limit === "") {
+          delete obj.config["ensembler"].docker_config.resource_request.cpu_limit;
+        }
       }
     }