test trainer image built with cpu

Signed-off-by: helenxie-bit <[email protected]>
kubeflow · Jan 23, 2025 · fca94ae · fca94ae
1 parent e5bf840
commit fca94ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
@@ -635,7 +635,7 @@ class name in this argument.
 
             container_spec = training_utils.get_container_spec(
                 name=JOB_PARAMETERS[PYTORCHJOB_KIND]["container"],
-                base_image=TRAINER_TRANSFORMER_IMAGE,
+                base_image="docker.io/helenxiehz428/trainer:test",
                 args=[
                     "--model_uri",
                     model_provider_parameters.model_uri,

diff --git a/test/e2e/v1beta1/scripts/gh-actions/run-e2e-tune-api.py b/test/e2e/v1beta1/scripts/gh-actions/run-e2e-tune-api.py
@@ -14,7 +14,7 @@
 from verify import verify_experiment_results
 
 # Experiment timeout is 40 min.
-EXPERIMENT_TIMEOUT = 60 * 10
+EXPERIMENT_TIMEOUT = 60 * 15
 
 # The default logging config.
 logging.basicConfig(level=logging.INFO)