hirundo 0.1.18__py3-none-any.whl → 0.2.3.post1__py3-none-any.whl

hirundo/cli.py

@@ -1,7 +1,6 @@
 import os
 import re
 import sys
-import typing
 from pathlib import Path
 from typing import Annotated
 from urllib.parse import urlparse
@@ -28,9 +27,7 @@ app = typer.Typer(
 )
 
 
-def _upsert_env(
-    dotenv_filepath: typing.Union[str, Path], var_name: str, var_value: str
-):
+def _upsert_env(dotenv_filepath: str | Path, var_name: str, var_value: str):
     """
     Change an environment variable in the .env file.
     If the variable does not exist, it will be added.
@@ -88,7 +85,7 @@ def setup_api_key(
     ],
 ):
     """
-    Setup the API key for the Hirundo client library.
+    Setup the API key for the Hirundo Python SDK.
     Values are saved to a .env file in the current directory for use by the library in requests.
     """
     saved_to = upsert_env("API_KEY", api_key)
@@ -115,7 +112,7 @@ def change_api_remote(
     ],
 ):
     """
-    Change the API server address for the Hirundo client library.
+    Change the API server address for the Hirundo Python SDK.
     This is the same address where you access the Hirundo web interface.
     """
     api_host = fix_api_host(api_host)
@@ -151,7 +148,7 @@ def setup(
     ],
 ):
     """
-    Setup the Hirundo client library.
+    Setup the Hirundo Python SDK.
     """
     api_host = fix_api_host(api_host)
     api_host_saved_to = upsert_env("API_HOST", api_host)
@@ -198,9 +195,9 @@ def check_run(
     """
     Check the status of a run.
     """
-    from hirundo.dataset_optimization import OptimizationDataset
+    from hirundo.dataset_qa import QADataset
 
-    results = OptimizationDataset.check_run_by_id(run_id)
+    results = QADataset.check_run_by_id(run_id)
     print(f"Run results saved to {results.cached_zip_path}")
 
 
@@ -209,9 +206,9 @@ def list_runs():
     """
     List all runs available.
     """
-    from hirundo.dataset_optimization import OptimizationDataset
+    from hirundo.dataset_qa import QADataset
 
-    runs = OptimizationDataset.list_runs()
+    runs = QADataset.list_runs()
 
     console = Console()
     table = Table(

hirundo/dataset_enum.py

@@ -24,6 +24,7 @@ class DatasetMetadataType(str, Enum):
     HIRUNDO_CSV = "HirundoCSV"
     COCO = "COCO"
     YOLO = "YOLO"
+    HuggingFaceAudio = "HuggingFaceAudio"
     KeylabsObjDetImages = "KeylabsObjDetImages"
     KeylabsObjDetVideo = "KeylabsObjDetVideo"
     KeylabsObjSegImages = "KeylabsObjSegImages"
@@ -44,3 +45,4 @@ class StorageTypes(str, Enum):
     """
     Local storage config is only supported for on-premises installations.
     """
+    HUGGINGFACE = "HuggingFace"