ml-analytics-tools 0.7.1__tar.gz → 0.7.2__tar.gz

{ml_analytics_tools-0.7.1 → ml_analytics_tools-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ml-analytics-tools
-Version: 0.7.1
+Version: 0.7.2
 Summary: Tools for ML projects and data management
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{ml_analytics_tools-0.7.1 → ml_analytics_tools-0.7.2}/ml_analytics/data_connector.py

@@ -678,10 +678,8 @@ class DataConnector:
                 self.cursor = self.connection.cursor()
                 self._start_idle_timer()  # Start idle timer on new connection
 
-                # Initialize the default S3 connector only when a default bucket is configured.
-                if self._s3_bucket:
-                    self.s3 = self._get_s3_for_bucket(self._s3_bucket)
-
+                # Defer S3 connector initialization until an S3 operation is actually required.
+                # This avoids AWS credential refresh/SSO login when only the database connection is used.
             except Exception as e:
                 log_and_raise_error(self._logger, f"Failed to connect to {self.engine.title()}: {e}")
 

{ml_analytics_tools-0.7.1 → ml_analytics_tools-0.7.2}/ml_analytics_tools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ml-analytics-tools
-Version: 0.7.1
+Version: 0.7.2
 Summary: Tools for ML projects and data management
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{ml_analytics_tools-0.7.1 → ml_analytics_tools-0.7.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ml-analytics-tools"
-version = "0.7.1"
+version = "0.7.2"
 description = "Tools for ML projects and data management"
 readme = "README.md"
 requires-python = ">=3.11"

{ml_analytics_tools-0.7.1 → ml_analytics_tools-0.7.2}/tests/test_db_s3.py

@@ -650,6 +650,23 @@ class TestThreadSafeConnect:
 
             assert mock_redshift.connect.call_count == 1
 
+    def test_connect_does_not_initialize_s3_when_default_bucket_configured(self, mock_credentials):
+        """connect() should not initialize S3 unless it's actually used."""
+        with patch("ml_analytics.data_connector.redshift_connector") as mock_redshift, patch(
+            "ml_analytics.s3_connector.boto3"
+        ) as mock_boto3:
+            mock_connection = MagicMock()
+            mock_connection.closed = False
+            mock_cursor = MagicMock()
+            mock_connection.cursor.return_value = mock_cursor
+            mock_redshift.connect.return_value = mock_connection
+
+            dc = DataConnector()
+            dc.connect()
+
+            assert dc.s3 is None
+            assert mock_boto3.Session.call_count == 0
+
 
 # === Tests for multi-statement SQL in unload_to_s3 ===