fix test_offloaded_literal

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Sep 19, 2024 · 6b59d89 · 6b59d89
1 parent be6c024
commit 6b59d89
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py
@@ -3218,10 +3218,10 @@ def test_union_file_directory():
         (typing.List[Color], [Color.RED, Color.GREEN, Color.BLUE]),
         (typing.List[Annotated[int, "tag"]], [1, 2, 3]),
         (typing.List[Annotated[str, "tag"]], ["a", "b", "c"]),
-        (typing.Dict[int, str], {"1": "a", "2": "b", "3": "c"}),
+        (typing.Dict[int, str], {1: "a", 2: "b", 3: "c"}),
         (typing.Dict[str, int], {"a": 1, "b": 2, "c": 3}),
         (typing.Dict[str, typing.List[int]], {"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]}),
-        (typing.Dict[str, typing.Dict[int, str]], {"a": {"1": "a", "2": "b", "3": "c"}, "b": {"4": "d", "5": "e", "6": "f"}}),
+        (typing.Dict[str, typing.Dict[int, str]], {"a": {1: "a", 2: "b", 3: "c"}, "b": {4: "d", 5: "e", 6: "f"}}),
         (typing.Union[int, str], 42),
         (typing.Union[int, str], "hello"),
         (typing.Union[typing.List[int], typing.List[str]], [1, 2, 3]),