datatoolpack 0.7.2__tar.gz → 0.7.4__tar.gz

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

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

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

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

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

@@ -445,8 +445,14 @@ class AutoDataClient:
             session_id:         Session to download.
             download_path:      Directory to save files (created if needed).
                                 Defaults to ``./auto_data_outputs/<session_id>/``.
-            output_preferences: Subset of filenames to include.  ``None``
-                                downloads everything.
+            output_preferences: Subset of outputs to include.  ``None``
+                                downloads everything.  Each entry can be
+                                either the full filename
+                                (``'dsg_output.csv'``) or the short key
+                                (``'dsg_output'``); the SDK appends ``.csv``
+                                to extensionless entries automatically.
+                                Use ``'pipeline_report'`` (or
+                                ``'pipeline_report.pdf'``) for the PDF report.
             compressed:         If ``True`` (default), downloads a single ZIP
                                 archive and extracts it.  If ``False``,
                                 downloads each file individually.
@@ -463,7 +469,29 @@ class AutoDataClient:
         if compressed:
             body: Dict = {}
             if output_preferences:
-                body["files"] = output_preferences
+                # Server <= 0.7.3 used exact filename match — a user passing
+                # the key-form name ('dsg_output' instead of 'dsg_output.csv')
+                # got back an empty archive. Server >= 0.7.4 accepts both
+                # forms, but to keep older deployments working we also
+                # canonicalize on the client: if an entry has no extension,
+                # append the conventional .csv (which is correct for every
+                # AutoData output except pipeline_report → .pdf and the
+                # *_artifacts.pkl files, which users rarely cherry-pick).
+                EXTENSIONLESS_DEFAULTS = {
+                    "pipeline_report": "pipeline_report.pdf",
+                }
+                normalized = []
+                for item in output_preferences:
+                    item = str(item).strip()
+                    if not item:
+                        continue
+                    if "." in item:
+                        normalized.append(item)
+                    elif item in EXTENSIONLESS_DEFAULTS:
+                        normalized.append(EXTENSIONLESS_DEFAULTS[item])
+                    else:
+                        normalized.append(f"{item}.csv")
+                body["files"] = normalized
             r = self._request(
                 "POST",
                 self._url(f"/download-archive/{session_id}"),
@@ -759,7 +787,7 @@ class AutoDataClient:
                   ``kafka``, ``kinesis``.
 
                 Generic:
-                  ``http`` (REST/HTTP API), ``mock`` (test).
+                  ``http`` (REST/HTTP API).
             secrets:        Inline connection secrets (e.g. connection_string,
                             bucket, access keys).
             credential_id:  ID of a saved credential on the server

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

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

{datatoolpack-0.7.2 → datatoolpack-0.7.4}/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.2",
+    version="0.7.4",
     description="Official Python SDK for the AutoData ML data preparation pipeline API",
     long_description=long_description,
     long_description_content_type="text/markdown",