Flake fix

flyteorg · Apr 16, 2021 · 7005af8 · 7005af8
1 parent f010509
commit 7005af8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flytekit/extras/sqlalchemy/task.py b/flytekit/extras/sqlalchemy/task.py
@@ -4,9 +4,10 @@
 import pandas as pd
 from sqlalchemy import create_engine
 
-from flytekit import current_context, kwtypes, Secret
+from flytekit import current_context, kwtypes
 from flytekit.core.base_sql_task import SQLTask
 from flytekit.core.python_function_task import PythonInstanceTask
+from flytekit.models.security import Secret
 from flytekit.types.schema import FlyteSchema