Fix dataclass transformer test

Signed-off-by: Eduardo Apolinario <[email protected]>
flyteorg · Nov 5, 2021 · 73c79cb · 73c79cb
1 parent c703c11
commit 73c79cb
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
@@ -434,10 +434,10 @@ def test_dataclass_transformer():
             "InnerstructSchema": {
                 "additionalProperties": False,
                 "properties": {
-                    "a": {"format": "integer", "title": "a", "type": "number"},
+                    "a": {"title": "a", "type": "integer"},
                     "b": {"default": None, "title": "b", "type": ["string", "null"]},
                     "c": {
-                        "items": {"format": "integer", "title": "c", "type": "number"},
+                        "items": {"title": "c", "type": "integer"},
                         "title": "c",
                         "type": "array",
                     },