unitlab 2.4.3__tar.gz → 2.4.5__tar.gz

{unitlab-2.4.3 → unitlab-2.4.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitlab
-Version: 2.4.3
+Version: 2.4.5
 Summary: Python SDK for the Unitlab.ai data annotation platform
 Project-URL: Homepage, https://unitlab.ai
 Project-URL: Documentation, https://docs.unitlab.ai

{unitlab-2.4.3 → unitlab-2.4.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "unitlab"
-version = "2.4.3"
+version = "2.4.5"
 description = "Python SDK for the Unitlab.ai data annotation platform"
 readme = "README.md"
 license = "MIT"

{unitlab-2.4.3 → unitlab-2.4.5}/src/unitlab/cli.py

@@ -37,12 +37,6 @@ class DownloadType(str, Enum):
     files = "files"
 
 
-class DatasetSplit(str, Enum):
-    train = "train"
-    validation = "validation"
-    test = "test"
-
-
 @app.command(help="Configure the credentials")
 def configure(
     api_key: Annotated[
@@ -124,11 +118,14 @@ def dataset_download(
         typer.Option(help="Export type (e.g. COCO, YOLOv8, YOLOv5, UUEF)"),
     ] = "COCO",
     split_type: Annotated[
-        DatasetSplit, typer.Option(help="Dataset split (train, validation, test)")
-    ] = DatasetSplit.train,
+        str | None,
+        typer.Option(
+            help="Dataset split (e.g. train, validation). Auto-detected if omitted."
+        ),
+    ] = None,
 ):
     if download_type == DownloadType.annotation:
-        get_client(api_key).dataset_download(str(pk), export_type, split_type.value)
+        get_client(api_key).dataset_download(str(pk), export_type, split_type)
     else:
         typer.echo(get_client(api_key).dataset_download_files(str(pk)))
 

{unitlab-2.4.3 → unitlab-2.4.5}/src/unitlab/client.py

@@ -309,16 +309,19 @@ class UnitlabClient:
         return self._get(f"/api/sdk/datasets/?pretty={pretty}")
 
     def dataset_download(
-        self, dataset_id: str, export_type: str, split_type: str
+        self,
+        dataset_id: str,
+        export_type: str,
+        split_type: str | None = None,
     ) -> str:
-        response = self._post(
-            f"/api/sdk/datasets/{dataset_id}/",
-            data={
-                "download_type": "annotation",
-                "export_type": export_type,
-                "split_type": split_type,
-            },
-        )
+        data: dict[str, Any] = {
+            "download_type": "annotation",
+            "export_type": export_type,
+        }
+        if split_type is not None:
+            data["split_type"] = split_type
+
+        response = self._post(f"/api/sdk/datasets/{dataset_id}/", data=data)
 
         file_url = response["file"]
         try:
@@ -397,7 +400,9 @@ class UnitlabClient:
 
         async def main() -> None:
             with tqdm.tqdm(total=len(files_to_download), ncols=80) as pbar:
-                async with httpx.AsyncClient(timeout=600.0) as client:
+                async with httpx.AsyncClient(
+                    timeout=600.0, follow_redirects=True
+                ) as client:
                     tasks = [
                         download_file(client=client, dataset_file=df)
                         for df in files_to_download