Fixed test

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Oct 25, 2021 · c6d219b · c6d219b
1 parent 994437a
commit c6d219b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/flytekit/common/translator.py b/flytekit/common/translator.py
@@ -120,7 +120,7 @@ def get_serializable_task(
     )
     if settings.should_fast_serialize() and isinstance(entity, PythonAutoContainerTask):
         entity.reset_command_fn()
-
+    print(tt)
     return task_models.TaskSpec(template=tt)
 
 

diff --git a/tests/flytekit/unit/core/test_type_hints.py b/tests/flytekit/unit/core/test_type_hints.py
@@ -1425,5 +1425,9 @@ def foo3(a: typing.Dict) -> typing.Dict:
     with pytest.raises(TypeError, match="Failed to convert return value for var o0 for function test_type_hints.foo2"):
         foo2(a=10, b="hello")
 
-    with pytest.raises(TypeError, match="Not a collection type simple: STRUCT\n but got a list \\[{'hello': 2}\\]"):
+    with pytest.raises(
+        TypeError,
+        match="Not a collection type simple: STRUCT\nmetadata {\n  fields {\n    key: 'python_class_name'\n"
+        "    value {\n      string_value: 'VT'\n    }\n  }\n}\n but got a list \\[{'hello': 2\\}]",
+    ):
         foo3(a=[{"hello": 2}])