Use a lightweight image for the Katib KFP trial

Signed-off-by: Stefano Fioravanzo <[email protected]> Reviewed-by: Ilias Katsakioris <[email protected]>
kubeflow-kale · Jun 13, 2020 · 56f9e5e · 56f9e5e
1 parent f20978e
commit 56f9e5e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/kale/rpc/katib.py b/backend/kale/rpc/katib.py
@@ -169,7 +169,7 @@ def create_katib_experiment(request, pipeline_id, pipeline_metadata,
     katib_spec = _sanitize_katib_spec(request, katib_spec)
 
     trial_parameters = {
-        "image": "gcr.io/arrikto/katib-kfp-trial:ba82feb",
+        "image": "gcr.io/arrikto/katib-kfp-trial:acbe872",
         "pipeline_id": pipeline_id,
         "experiment_name": pipeline_metadata.get(
             "experiment_name")}

diff --git a/backend/kale/tests/assets/yamls/katib-experiment.yaml b/backend/kale/tests/assets/yamls/katib-experiment.yaml
@@ -40,7 +40,7 @@ spec:
         \  namespace: {{.NameSpace}}\nspec:\n  backoffLimit: 0\n  template:\n    metadata:\n\
         \      annotations:\n        sidecar.istio.io/inject: \"false\"\n    spec:\n\
         \      restartPolicy: Never\n      containers:\n        - name: {{.Trial}}\n\
-        \          image: gcr.io/arrikto/katib-kfp-trial:ba82feb\n          command:\n\
+        \          image: gcr.io/arrikto/katib-kfp-trial:acbe872\n          command:\n\
         \            - python3 -u -c \"from kale.utils.kfp_utils                import\
         \ create_and_wait_kfp_run;                create_and_wait_kfp_run(       \
         \             pipeline_id='12345',                    run_name='{{.Trial}}',\