diff --git a/pkg/components/estimator/estimator.go b/pkg/components/estimator/estimator.go index 87970408..af27507c 100644 --- a/pkg/components/estimator/estimator.go +++ b/pkg/components/estimator/estimator.go @@ -28,7 +28,7 @@ import ( const ( // NOTE: update tests/images.yaml when changing this image - StableImage = "quay.io/sustainable_computing_io/kepler_model_server:v0.7.11" + StableImage = "quay.io/sustainable_computing_io/kepler_model_server:v0.7.11-2" waitForSocketCommand = "until [ -e /tmp/estimator.sock ]; do sleep 1; done && %s" ) @@ -76,8 +76,8 @@ func Container(image string) corev1.Container { ImagePullPolicy: corev1.PullIfNotPresent, Name: "estimator", VolumeMounts: mounts, - Command: []string{"python3.10"}, - Args: []string{"-u", "src/estimate/estimator.py"}, + Command: []string{"estimator"}, + Args: []string{"-l", "info"}, } } diff --git a/pkg/components/modelserver/modelserver.go b/pkg/components/modelserver/modelserver.go index a35b0acb..57d4a5ad 100644 --- a/pkg/components/modelserver/modelserver.go +++ b/pkg/components/modelserver/modelserver.go @@ -38,7 +38,7 @@ const ( const ( defaultModelServer = "http://%s.%s.svc.cluster.local:%d" - StableImage = "quay.io/sustainable_computing_io/kepler_model_server:v0.7.11" + StableImage = "quay.io/sustainable_computing_io/kepler_model_server:v0.7.11-2" ) var ( @@ -86,8 +86,8 @@ func NewDeployment(deployName string, ms *v1alpha1.InternalModelServerSpec, name Name: "http", }}, VolumeMounts: mounts, - Command: []string{"python3.10"}, - Args: []string{"-u", "src/server/model_server.py"}, + Command: []string{"model-server"}, + Args: []string{"-l", "info"}, }} return &appsv1.Deployment{ diff --git a/tests/images.yaml b/tests/images.yaml index f61a7248..ab965538 100644 --- a/tests/images.yaml +++ b/tests/images.yaml @@ -1,3 +1,3 @@ images: - component: 'model-server' - image: 'quay.io/sustainable_computing_io/kepler_model_server:v0.7.11' + image: 'quay.io/sustainable_computing_io/kepler_model_server:v0.7.11-2'