atlasai-dstoolkit-client 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

atlasai/toolkit/feature.py

@@ -48,7 +48,7 @@ class FeatureExport:
         return self._search.search
 
     def details(self):
-        if self.export is None or self.export.status not in ['completed', 'failed']:
+        if self.export is None or self.export.get('status') not in ['succeeded', 'failed']:
             self._export = self._details()
         return self.export
 
@@ -57,10 +57,15 @@ class FeatureExport:
         return self.export
 
     def results(self, limit=None) -> pd.DataFrame:
-        if self.export.status not in ['completed', 'failed']:
-            raise Exception(f'Export state is: {self.export.status}')
-
-        path = getattr(self.export, 'url_path', getattr(self.export, 'output_path'))
+        if not self.export:
+            return pd.DataFrame([])
+
+        if self.export.get('status') not in ['succeeded', 'failed']:
+            raise Exception(f'Export state is: {self.export.get("status")}')
+        path = None
+        grouped = self.export.get('search_result', {}).get('grouped', [])
+        if grouped:
+            path = grouped[0]['url']
         if not path:
             raise Exception('Path not found')
 
@@ -70,6 +75,6 @@ class FeatureExport:
         return df
 
     def _details(self):
-        resource = f'feature/export/{self._search.id}/details'
+        resource = f'feature/export/{self.id}/details'
         _, data = api._get(resource=resource)
         return data['data']

{atlasai_dstoolkit_client-0.0.4.dist-info → atlasai_dstoolkit_client-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atlasai-dstoolkit-client
-Version: 0.0.4
+Version: 0.0.6
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: AtlasAI SWE
@@ -12,6 +12,7 @@ License-File: LICENSE.txt
 Requires-Dist: arrow (<2.0.0,>=1.3.0)
 Requires-Dist: furl (<3.0.0,>=2.1.2)
 Requires-Dist: pandas (<3.0.0,>=2.2.3)
+Requires-Dist: pyarrow (>=19.0.1)
 Requires-Dist: python-dotenv (<2.0.0,>=1.0.1)
 Requires-Dist: requests (<=3.0.0,>=2.32.3)
 Requires-Dist: semver (<4.0.0,>=3.0.2)

{atlasai_dstoolkit_client-0.0.4.dist-info → atlasai_dstoolkit_client-0.0.6.dist-info}/RECORD

@@ -2,13 +2,13 @@ atlasai/toolkit/__init__.py,sha256=szu5LHZ41ccWztCI2LvQ3QnZRybosgTbU3yd6tS7cUw,9
 atlasai/toolkit/api.py,sha256=BvO-gLRmbmkKduwbbADjcLlIkS9blzfM_cbMR4DhQmU,5269
 atlasai/toolkit/constants.py,sha256=sE0PeFa9_htCPVFADHbkyPIz3QOai0tAIA5IiiZI8wA,329
 atlasai/toolkit/fabric.py,sha256=6aFR2PGQc9P3Qa07WdBg9eKoUzU8n2y_-gGjYcyMrWY,1921
-atlasai/toolkit/feature.py,sha256=EM0shPWhq11CK5kGnu5Wq9-UAWVsw5PIXRcmC27Yc4A,2275
+atlasai/toolkit/feature.py,sha256=f75nlELRgsVUYaVMKl055Q4ob3C-eVGp6xJHVzohsJk,2418
 atlasai/toolkit/init.py,sha256=JkdJ6QGdYWrq65jgz2pn5RYXUeUe2Ez88_-eMf5CNi0,1100
 atlasai/toolkit/login.py,sha256=n4ydfo9qCsmbZq6er1xeljBD76vdTJGjbhYHMmOyDbQ,3061
 atlasai/toolkit/requests.py,sha256=X86nIo07hAjUlilZcZ1lV8RB7KOsTKbTGtcY_SpFEXY,1223
 atlasai/toolkit/utils.py,sha256=lYh3P2XOshRgHCjFeXJ0FOJWQW64sddgx8c2kL6Wqwc,1566
-atlasai_dstoolkit_client-0.0.4.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-atlasai_dstoolkit_client-0.0.4.dist-info/METADATA,sha256=5Pa19r85pmM0fCjrHIeNUmwpYiqDAyLFEvSsFjjptt0,1301
-atlasai_dstoolkit_client-0.0.4.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-atlasai_dstoolkit_client-0.0.4.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
-atlasai_dstoolkit_client-0.0.4.dist-info/RECORD,,
+atlasai_dstoolkit_client-0.0.6.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+atlasai_dstoolkit_client-0.0.6.dist-info/METADATA,sha256=2ct5cuCh8aGH8sTUCxuVj6jaieu8kQXjPuAhUVRbF88,1335
+atlasai_dstoolkit_client-0.0.6.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+atlasai_dstoolkit_client-0.0.6.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
+atlasai_dstoolkit_client-0.0.6.dist-info/RECORD,,