TF2 instead of TF1 as image for k8s tests.

determined-ai · Jan 5, 2024 · b71a8fc · b71a8fc
1 parent 8f79f91
commit b71a8fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/e2e_tests/tests/command/test_run.py b/e2e_tests/tests/command/test_run.py
@@ -551,7 +551,7 @@ def test_k8_init_containers(using_k8s: bool) -> None:
                 "spec": {
                     "initContainers": [
                         {
-                            "image": conf.TF1_CPU_IMAGE,
+                            "image": conf.TF2_CPU_IMAGE,
                             "name": "simple-init-container",
                             "command": ["/bin/bash"],
                             "args": ["-c", "exit 1"],
@@ -581,7 +581,7 @@ def test_k8_sidecars(using_k8s: bool) -> None:
                 "spec": {
                     "containers": [
                         {
-                            "image": conf.TF1_CPU_IMAGE,
+                            "image": conf.TF2_CPU_IMAGE,
                             "name": "sidecar",
                             "command": ["/bin/bash"],
                         }