Fixed lint

Signed-off-by: Kevin Su <pingsutw@apache.org>
flyteorg · Oct 26, 2021 · 8704d7c · 8704d7c
1 parent f59afa5
commit 8704d7c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/flytekit/types/schema/types.py b/flytekit/types/schema/types.py
@@ -9,7 +9,7 @@
 from typing import Type
 
 import numpy as _np
-from dataclasses_json import dataclass_json, config
+from dataclasses_json import config, dataclass_json
 from marshmallow import fields
 
 from flytekit.core.context_manager import FlyteContext, FlyteContextManager
@@ -172,7 +172,9 @@ def get_handler(cls, t: Type) -> SchemaHandler:
 @dataclass_json
 @dataclass
 class FlyteSchema(object):
-    supported_mode: typing.Optional[str] = field(default=SchemaOpenMode.WRITE, metadata=config(mm_field=fields.String()))
+    supported_mode: typing.Optional[str] = field(
+        default=SchemaOpenMode.WRITE, metadata=config(mm_field=fields.String())
+    )
     local_path: typing.Optional[str] = field(default=None, metadata=config(mm_field=fields.String()))
     remote_path: typing.Optional[str] = field(default=None, metadata=config(mm_field=fields.String()))
     """

diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py
@@ -14,7 +14,7 @@
 from marshmallow_jsonschema import JSONSchema
 
 import flytekit.models.core.types
-from flytekit import workflow, task, kwtypes
+from flytekit import kwtypes, task, workflow
 from flytekit.common.exceptions import user as user_exceptions
 from flytekit.core.context_manager import FlyteContext, FlyteContextManager
 from flytekit.core.type_engine import (