diff --git a/flytekit/extras/sqlalchemy/task.py b/flytekit/extras/sqlalchemy/task.py index 7def055c64b..ee38260e25a 100644 --- 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