change image name

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Apr 9, 2024 · 20fb365 · 20fb365
1 parent 902e67a
commit 20fb365
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/flytekit/unit/core/test_local_raw_container.py b/tests/flytekit/unit/core/test_local_raw_container.py
@@ -126,7 +126,7 @@ def test_primitive_types_wf():
         output_data_dir="/var/outputs",
         inputs=kwtypes(a=int, b=bool, c=float, d=str, e=datetime.datetime, f=datetime.timedelta),
         outputs=kwtypes(a=int, b=bool, c=float, d=str, e=datetime.datetime, f=datetime.timedelta),
-        image="futureoutlier/rawcontainer:0320",
+        image="flytekit:rawcontainer",
         command=[
             "python",
             "return_same_value.py",
@@ -180,7 +180,7 @@ def test_input_output_dir_manipulation():
         output_data_dir="/outputs",
         inputs=kwtypes(a=int, b=bool, c=float, d=str, e=datetime.datetime, f=datetime.timedelta),
         outputs=kwtypes(a=int, b=bool, c=float, d=str, e=datetime.datetime, f=datetime.timedelta),
-        image="futureoutlier/rawcontainer:0320",
+        image="flytekit:rawcontainer",
         command=[
             "python",
             "return_same_value.py",