flytekitplugins-sqlalchemy 1.10.2__tar.gz → 1.10.3__tar.gz

{flytekitplugins-sqlalchemy-1.10.2 → flytekitplugins-sqlalchemy-1.10.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flytekitplugins-sqlalchemy
-Version: 1.10.2
+Version: 1.10.3
 Summary: SQLAlchemy plugin for flytekit
 Author: dolthub
 Author-email: max@dolthub.com
@@ -19,3 +19,4 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.8
 Requires-Dist: flytekit<2.0.0,>=1.3.0b2
 Requires-Dist: sqlalchemy>=1.4.7
+Requires-Dist: pandas

{flytekitplugins-sqlalchemy-1.10.2 → flytekitplugins-sqlalchemy-1.10.3}/flytekitplugins/sqlalchemy/task.py

@@ -1,11 +1,7 @@
 import typing
 from dataclasses import dataclass
 
-import pandas as pd
-from pandas.io.sql import pandasSQL_builder
-from sqlalchemy import create_engine, text  # type: ignore
-
-from flytekit import current_context, kwtypes
+from flytekit import current_context, kwtypes, lazy_module
 from flytekit.configuration import SerializationSettings
 from flytekit.configuration.default_images import DefaultImages, PythonVersion
 from flytekit.core.base_sql_task import SQLTask
@@ -16,6 +12,10 @@ from flytekit.models import task as task_models
 from flytekit.models.security import Secret
 from flytekit.types.schema import FlyteSchema
 
+pd = lazy_module("pandas")
+pandas_io_sql = lazy_module("pandas.io.sql")
+sqlalchemy = lazy_module("sqlalchemy")
+
 
 class SQLAlchemyDefaultImages(DefaultImages):
     """Default images for the sqlalchemy flytekit plugin."""
@@ -126,7 +126,7 @@ class SQLAlchemyTaskExecutor(ShimTaskExecutor[SQLAlchemyTask]):
                 value = current_context().secrets.get(group=secret_dict["group"], key=secret_dict["key"])
                 tt.custom["connect_args"][key] = value
 
-        engine = create_engine(tt.custom["uri"], connect_args=tt.custom["connect_args"], echo=False)
+        engine = sqlalchemy.create_engine(tt.custom["uri"], connect_args=tt.custom["connect_args"], echo=False)
         logger.info(f"Connecting to db {tt.custom['uri']}")
 
         interpolated_query = SQLAlchemyTask.interpolate_query(tt.custom["query_template"], **kwargs)
@@ -134,7 +134,7 @@ class SQLAlchemyTaskExecutor(ShimTaskExecutor[SQLAlchemyTask]):
         with engine.begin() as connection:
             df = None
             if tt.interface.outputs:
-                df = pd.read_sql_query(text(interpolated_query), connection)
+                df = pd.read_sql_query(sqlalchemy.text(interpolated_query), connection)
             else:
-                pandasSQL_builder(connection).execute(text(interpolated_query))
+                pandas_io_sql.pandasSQL_builder(connection).execute(sqlalchemy.text(interpolated_query))
         return df

{flytekitplugins-sqlalchemy-1.10.2 → flytekitplugins-sqlalchemy-1.10.3}/flytekitplugins_sqlalchemy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flytekitplugins-sqlalchemy
-Version: 1.10.2
+Version: 1.10.3
 Summary: SQLAlchemy plugin for flytekit
 Author: dolthub
 Author-email: max@dolthub.com
@@ -19,3 +19,4 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.8
 Requires-Dist: flytekit<2.0.0,>=1.3.0b2
 Requires-Dist: sqlalchemy>=1.4.7
+Requires-Dist: pandas

{flytekitplugins-sqlalchemy-1.10.2 → flytekitplugins-sqlalchemy-1.10.3}/flytekitplugins_sqlalchemy.egg-info/requires.txt

@@ -1,2 +1,3 @@
 flytekit<2.0.0,>=1.3.0b2
 sqlalchemy>=1.4.7
+pandas

{flytekitplugins-sqlalchemy-1.10.2 → flytekitplugins-sqlalchemy-1.10.3}/setup.py

@@ -4,9 +4,9 @@ PLUGIN_NAME = "sqlalchemy"
 
 microlib_name = f"flytekitplugins-{PLUGIN_NAME}"
 
-plugin_requires = ["flytekit>=1.3.0b2,<2.0.0", "sqlalchemy>=1.4.7"]
+plugin_requires = ["flytekit>=1.3.0b2,<2.0.0", "sqlalchemy>=1.4.7", "pandas"]
 
-__version__ = "1.10.2"
+__version__ = "1.10.3"
 
 setup(
     name=microlib_name,

{flytekitplugins-sqlalchemy-1.10.2 → flytekitplugins-sqlalchemy-1.10.3}/tests/test_task.py

@@ -205,7 +205,7 @@ def test_task_serialization_deserialization_with_secret(sql_server):
     assert r.iat[0, 0] == 1
 
 
-@mock.patch("flytekit.clis.sdk_in_container.helpers.FlyteRemote", spec=FlyteRemote)
+@mock.patch("flytekit.configuration.plugin.FlyteRemote", spec=FlyteRemote)
 @mock.patch("flytekit.clients.friendly.SynchronousFlyteClient", spec=SynchronousFlyteClient)
 def test_register_sql_task(mock_client, mock_remote):
     mock_remote._client = mock_client