datatoolpack 0.7.4__tar.gz → 0.8.0__tar.gz

{datatoolpack-0.7.4 → datatoolpack-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatoolpack
-Version: 0.7.4
+Version: 0.8.0
 Summary: Official Python SDK for the AutoData ML data preparation pipeline API
 Home-page: https://autodata.datatoolpack.com
 Author: AutoData Team

{datatoolpack-0.7.4 → datatoolpack-0.8.0}/datatoolpack/__init__.py

@@ -1,4 +1,4 @@
 from .client import AutoDataClient, AutoDataError
 
-__version__ = "0.7.4"
+__version__ = "0.8.0"
 __all__ = ["AutoDataClient", "AutoDataError", "__version__"]

{datatoolpack-0.7.4 → datatoolpack-0.8.0}/datatoolpack/client.py

@@ -74,7 +74,7 @@ class AutoDataClient:
     Args:
         api_key:      API key string starting with ``dtpk_``.
         base_url:     Base URL of the AutoData server (no trailing slash).
-        timeout:      HTTP request timeout in seconds (default 120).
+        timeout:      HTTP request timeout in seconds (default 300).
         max_retries:  Max automatic retries for transient errors
                       (429 / 502 / 503 / 504).  Default 3.
 
@@ -89,7 +89,7 @@ class AutoDataClient:
         api_key: Optional[str] = None,
         passcode: Optional[str] = None,
         base_url: Optional[str] = None,
-        timeout: int = 120,
+        timeout: int = 300,
         max_retries: int = 3,
     ):
         # Resolve from env vars if not provided explicitly
@@ -195,7 +195,10 @@ class AutoDataClient:
                     return response  # let caller handle the error
                 retry_after = int(response.headers.get("Retry-After", 0))
                 wait = max(retry_after, 2 ** attempt)
-            except requests.ConnectionError as exc:
+            except (requests.ConnectionError, requests.Timeout) as exc:
+                # Retry transient read/connect timeouts too: a long-running
+                # server-side stage can briefly delay a poll past the per-request
+                # timeout; that should not fail the whole job.
                 last_exc = exc
                 if attempt == self.max_retries:
                     raise
@@ -364,7 +367,7 @@ class AutoDataClient:
         session_id: str,
         poll_interval: int = 2,
         timeout: Optional[float] = None,
-        stall_timeout: float = 600.0,
+        stall_timeout: float = 3600.0,
     ) -> Dict:
         """Block until a session reaches *completed*, *error*, or *cancelled*.
 
@@ -405,6 +408,8 @@ class AutoDataClient:
                 raise AutoDataError(f"Processing failed: {msg}")
             if status == "cancelled":
                 raise AutoDataError("Processing was cancelled")
+            if status == "failed":
+                raise AutoDataError(f"Processing failed: {msg}")
 
             # Detect stalled workers: status stays 'unknown'/'processing' with
             # no progress change. Prevents infinite loops when cpu_worker died.

{datatoolpack-0.7.4 → datatoolpack-0.8.0}/datatoolpack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatoolpack
-Version: 0.7.4
+Version: 0.8.0
 Summary: Official Python SDK for the AutoData ML data preparation pipeline API
 Home-page: https://autodata.datatoolpack.com
 Author: AutoData Team

{datatoolpack-0.7.4 → datatoolpack-0.8.0}/setup.py

@@ -7,7 +7,7 @@ with open(os.path.join(here, "README.md"), encoding="utf-8") as f:
 
 setup(
     name="datatoolpack",
-    version="0.7.4",
+    version="0.8.0",
     description="Official Python SDK for the AutoData ML data preparation pipeline API",
     long_description=long_description,
     long_description_content_type="text/markdown",