Fix test bug

Signed-off-by: Max Hoffman <[email protected]>
flyteorg · Apr 21, 2021 · da8ed8b · da8ed8b
1 parent 5318608
commit da8ed8b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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():