Fix a newly added test from master

Signed-off-by: Thomas Newton <[email protected]>
flyteorg · Dec 20, 2023 · 62c5209 · 62c5209
1 parent 8d1e734
commit 62c5209
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/flytekit/unit/core/test_utils.py b/tests/flytekit/unit/core/test_utils.py
@@ -1,3 +1,5 @@
+from collections import OrderedDict
+
 import pytest
 
 import flytekit
@@ -90,13 +92,13 @@ def t() -> str:
     assert t() == "hello world"
     assert t.get_config(settings=ss) == {}
 
-    ts = get_serializable_task(ss, t)
+    ts = get_serializable_task(OrderedDict(), ss, t)
     assert ts.template.config == {"foo": "bar"}
 
     @task
     @my_decorator
     def t() -> str:
         return "hello world"
 
-    ts = get_serializable_task(ss, t)
+    ts = get_serializable_task(OrderedDict(), ss, t)
     assert ts.template.config == {"foo": "baz"}