diff --git a/ui/src/services/router/TuringRouter.js b/ui/src/services/router/TuringRouter.js index ef1aea7bb..745b33f87 100644 --- 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; + } } }