diff --git a/tests/flytekit/unit/core/test_utils.py b/tests/flytekit/unit/core/test_utils.py index 910156f50a..ca0d07565d 100644 --- 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,7 +92,7 @@ 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 @@ -98,5 +100,5 @@ def t() -> str: 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"}