diff --git a/plugins/tests/sqlalchemy/test_task.py b/plugins/tests/sqlalchemy/test_task.py index a1258d44f3e..6adeca75197 100644 --- a/plugins/tests/sqlalchemy/test_task.py +++ b/plugins/tests/sqlalchemy/test_task.py @@ -11,6 +11,14 @@ from flytekit import kwtypes, task, workflow from flytekit.types.schema import FlyteSchema +tk = SQLAlchemyTask( + "test", + query_template="select * from tracks", + task_config=SQLAlchemyConfig( + uri="sqlite://", + ), +) + @pytest.fixture(scope="function") def sql_server():