hirundo 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

hirundo/__init__.py

@@ -32,4 +32,4 @@ __all__ = [
     "StorageIntegration",
 ]
 
-__version__ = "0.1.4"
+__version__ = "0.1.6"

hirundo/dataset_optimization.py

@@ -1,9 +1,11 @@
 import json
 import logging
 from collections.abc import AsyncGenerator, Generator
+from io import StringIO
 from typing import Union
 
 import httpx
+import pandas as pd
 import requests
 from pydantic import BaseModel, Field, model_validator
 
@@ -234,6 +236,13 @@ class OptimizationDataset(BaseModel):
         self.dataset_id = None
         self.run_id = None
 
+    @staticmethod
+    def _read_csv_to_df(data: dict):
+        if data["state"] == "SUCCESS":
+            data["result"] = pd.read_csv(StringIO(data["result"]))
+        else:
+            pass
+
     @staticmethod
     def check_run_by_id(run_id: str, retry=0) -> Generator[dict, None, None]:
         """
@@ -272,7 +281,11 @@ class OptimizationDataset(BaseModel):
                     sse.retry,
                 )
                 last_event = json.loads(sse.data)
-                yield last_event["data"]
+                if not last_event:
+                    continue
+                data = last_event["data"]
+                OptimizationDataset._read_csv_to_df(data)
+                yield data
         if not last_event or last_event["data"]["state"] == "PENDING":
             OptimizationDataset.check_run_by_id(run_id, retry + 1)
 

{hirundo-0.1.4.dist-info → hirundo-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hirundo
-Version: 0.1.4
+Version: 0.1.6
 Summary: This package is used to interface with Hirundo's platform. It provides a simple API to optimize your ML datasets.
 Author-email: Hirundo <dev@hirundo.io>
 License: MIT License
@@ -21,41 +21,42 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: pyyaml >=6.0.1
-Requires-Dist: types-PyYAML >=6.0.12
-Requires-Dist: pydantic >=2.7.1
-Requires-Dist: twine >=5.0.0
-Requires-Dist: python-dotenv >=1.0.1
-Requires-Dist: types-requests >=2.31.0
-Requires-Dist: typer >=0.12.3
-Requires-Dist: httpx >=0.27.0
-Requires-Dist: stamina >=24.2.0
-Requires-Dist: httpx-sse >=0.4.0
+Requires-Dist: pyyaml>=6.0.1
+Requires-Dist: types-PyYAML>=6.0.12
+Requires-Dist: pydantic>=2.7.1
+Requires-Dist: twine>=5.0.0
+Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: types-requests>=2.31.0
+Requires-Dist: typer>=0.12.3
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: stamina>=24.2.0
+Requires-Dist: httpx-sse>=0.4.0
+Requires-Dist: pandas>=2.2.2
 Provides-Extra: dev
-Requires-Dist: pyyaml >=6.0.1 ; extra == 'dev'
-Requires-Dist: types-PyYAML >=6.0.12 ; extra == 'dev'
-Requires-Dist: pydantic >=2.7.1 ; extra == 'dev'
-Requires-Dist: twine >=5.0.0 ; extra == 'dev'
-Requires-Dist: python-dotenv >=1.0.1 ; extra == 'dev'
-Requires-Dist: types-requests >=2.31.0 ; extra == 'dev'
-Requires-Dist: types-setuptools >=69.5.0 ; extra == 'dev'
-Requires-Dist: typer >=0.12.3 ; extra == 'dev'
-Requires-Dist: httpx >=0.27.0 ; extra == 'dev'
-Requires-Dist: stamina >=24.2.0 ; extra == 'dev'
-Requires-Dist: httpx-sse >=0.4.0 ; extra == 'dev'
-Requires-Dist: pytest >=8.2.0 ; extra == 'dev'
-Requires-Dist: pytest-asyncio >=0.23.6 ; extra == 'dev'
-Requires-Dist: uv ; extra == 'dev'
-Requires-Dist: pre-commit >=3.7.1 ; extra == 'dev'
-Requires-Dist: ruff ; extra == 'dev'
-Requires-Dist: bumpver ; extra == 'dev'
+Requires-Dist: pyyaml>=6.0.1; extra == "dev"
+Requires-Dist: types-PyYAML>=6.0.12; extra == "dev"
+Requires-Dist: pydantic>=2.7.1; extra == "dev"
+Requires-Dist: twine>=5.0.0; extra == "dev"
+Requires-Dist: python-dotenv>=1.0.1; extra == "dev"
+Requires-Dist: types-requests>=2.31.0; extra == "dev"
+Requires-Dist: types-setuptools>=69.5.0; extra == "dev"
+Requires-Dist: typer>=0.12.3; extra == "dev"
+Requires-Dist: httpx>=0.27.0; extra == "dev"
+Requires-Dist: stamina>=24.2.0; extra == "dev"
+Requires-Dist: httpx-sse>=0.4.0; extra == "dev"
+Requires-Dist: pytest>=8.2.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23.6; extra == "dev"
+Requires-Dist: uv; extra == "dev"
+Requires-Dist: pre-commit>=3.7.1; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Requires-Dist: bumpver; extra == "dev"
 Provides-Extra: docs
-Requires-Dist: sphinx >=7.4.7 ; extra == 'docs'
-Requires-Dist: sphinx-autobuild >=2024.4.16 ; extra == 'docs'
-Requires-Dist: sphinx-click >=5.0.1 ; extra == 'docs'
-Requires-Dist: autodoc-pydantic >=2.2.0 ; extra == 'docs'
-Requires-Dist: furo ; extra == 'docs'
-Requires-Dist: sphinx-multiversion ; extra == 'docs'
+Requires-Dist: sphinx>=7.4.7; extra == "docs"
+Requires-Dist: sphinx-autobuild>=2024.4.16; extra == "docs"
+Requires-Dist: sphinx-click>=5.0.1; extra == "docs"
+Requires-Dist: autodoc-pydantic>=2.2.0; extra == "docs"
+Requires-Dist: furo; extra == "docs"
+Requires-Dist: sphinx-multiversion; extra == "docs"
 
 # Hirundo client
 
@@ -65,6 +66,8 @@ This repo contains the source code for the Hirundo client library
 
 To learn about how to use this library, please visit the [http://docs.hirundo.io/](documentation) or see the Google Colab examples.
 
+Note: Currently we only support the main CPython release 3.9, 3.10 and 3.11. PyPy support may be introduced in the future.
+
 ## Development:
 
 ### Install dev dependencies

{hirundo-0.1.4.dist-info → hirundo-0.1.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-hirundo/__init__.py,sha256=GArvHFmbEOlyod-0VByoQNT_9CLgwf69t1wJVigoyzo,707
+hirundo/__init__.py,sha256=WeIIvtMy9ghxUKu1JyLcnVw1T1iQ9flQLDYpfqxb1g0,707
 hirundo/__main__.py,sha256=wcCrL4PjG51r5wVKqJhcoJPTLfHW0wNbD31DrUN0MWI,28
 hirundo/_constraints.py,sha256=-RAUV9GnCsaT9pLGSqYglKOeK0joPBBexGTo87j5nkI,425
 hirundo/_env.py,sha256=aObkRVLo9NBZiByd2FcoLrk3m8tnswuYzP4Tnj3EE-o,268
@@ -6,13 +6,13 @@ hirundo/_headers.py,sha256=htxHRjtD91C5D0svyk-zqhKV9LwQCEZauIa4ZTAfe5k,188
 hirundo/_iter_sse_retrying.py,sha256=WLp_lw8ycBuAxoJkkGBu4y74Ajhcu11r1X-vd5_571A,3352
 hirundo/_timeouts.py,sha256=IfX8-mrLp809-A_xSLv1DhIqZnO-Qvy4FcTtOtvqLog,42
 hirundo/cli.py,sha256=qj1Txt6lOU3V10SLtzH4uEWJ4DdkdOIEQaKn8wiJMss,3922
-hirundo/dataset_optimization.py,sha256=CLo9eclW_trDwzfr6uZlJ8JQb6XpWcKtACqSTaAF_fo,14583
+hirundo/dataset_optimization.py,sha256=aeEbd5J9Xo8l_keYXQ8v26JfYI-6VSZveRgCQN9H11E,14970
 hirundo/enum.py,sha256=-3w09g-_yRYIMiM8VA_Nb07WoQXf5IjyERTGonzNDs0,457
 hirundo/git.py,sha256=GtowxPL78KleVhSY3QISu7-cUPrFbWC4YWBAuzuzryw,4731
 hirundo/storage.py,sha256=CxRdSnZGf4mtzNV2Ge_hwowd9pDP7NT9-xvWTbl187M,8185
-hirundo-0.1.4.dist-info/LICENSE,sha256=fusGGjqT2RGlU6kbkaOk7d-gDnsjk17wq67AO0mwBZI,1065
-hirundo-0.1.4.dist-info/METADATA,sha256=Vtt_XENI41nDyygK5kCMOaAGGQK0noCYdR6eJ6LEeE0,4428
-hirundo-0.1.4.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-hirundo-0.1.4.dist-info/entry_points.txt,sha256=4ZtnA_Nl1Af8fLnHp3lwjbGDEGU1S6ujb_JwtuQ7ZPM,44
-hirundo-0.1.4.dist-info/top_level.txt,sha256=cmyNqrNZOAYxnywJGFI1AJBLe4SkH8HGsfFx6ncdrbI,8
-hirundo-0.1.4.dist-info/RECORD,,
+hirundo-0.1.6.dist-info/LICENSE,sha256=fusGGjqT2RGlU6kbkaOk7d-gDnsjk17wq67AO0mwBZI,1065
+hirundo-0.1.6.dist-info/METADATA,sha256=ILQyVMbnhMd7DuIw6RLldrxCbKI7pkrF46uE6TFesLQ,4530
+hirundo-0.1.6.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+hirundo-0.1.6.dist-info/entry_points.txt,sha256=4ZtnA_Nl1Af8fLnHp3lwjbGDEGU1S6ujb_JwtuQ7ZPM,44
+hirundo-0.1.6.dist-info/top_level.txt,sha256=cmyNqrNZOAYxnywJGFI1AJBLe4SkH8HGsfFx6ncdrbI,8
+hirundo-0.1.6.dist-info/RECORD,,