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

atlasai/toolkit/feature.py

@@ -12,11 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
 import uuid
+
 import pandas as pd
+from shapely import wkb
 
 from . import api, constants, utils
 
+logger = logging.getLogger(__name__)
 
 def export(*args, **kwargs):
     fe = FeatureExport(*args, **kwargs)
@@ -33,7 +37,7 @@ class FeatureExport:
         return f'FeatureExport({self._search.id})'
 
     def __str__(self):
-        return f'FeatureExport({self._search.id})'
+        return f'FeatureExport({self.id})'
 
     @property
     def page(self):
@@ -60,18 +64,29 @@ class FeatureExport:
         if not self.export:
             return pd.DataFrame([])
 
+        dfs = []
         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')
+        for group in grouped:
+            path = group.get('url')
+            if not path:
+                continue
+            df = pd.read_parquet(path, engine='pyarrow')
+            dfs.append(df)
+        if not dfs:
+            raise Exception('No dataframe found')
+
+        df = pd.concat(dfs, ignore_index=True) if len(dfs) > 1 else dfs[0]
 
-        df = pd.read_parquet(path, engine='pyarrow')
         if limit:
             df = df.head(limit)
+
+        try:
+            df['shape'] = wkb.loads(df['shape'])
+        except Exception:
+            logger.warning('Shape field is not in WKB format. Skipping conversion.')
+
         return df
 
     def _details(self):

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atlasai-dstoolkit-client
-Version: 0.0.6
+Version: 0.0.9
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: AtlasAI SWE
@@ -16,6 +16,7 @@ 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)
+Requires-Dist: shapely (~=2.1.0)
 Provides-Extra: dev
 Requires-Dist: coverage ; extra == 'dev'
 Requires-Dist: flake8 ; extra == 'dev'

{atlasai_dstoolkit_client-0.0.6.dist-info → atlasai_dstoolkit_client-0.0.9.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=f75nlELRgsVUYaVMKl055Q4ob3C-eVGp6xJHVzohsJk,2418
+atlasai/toolkit/feature.py,sha256=VKwX_yAwwMo1WuLZ_RbKVJEH4L1PAoYEPzWa0lH9ats,2828
 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.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,,
+atlasai_dstoolkit_client-0.0.9.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+atlasai_dstoolkit_client-0.0.9.dist-info/METADATA,sha256=ygcwrboXNHtv0Y4o7PrEcs2RZDcjkFfQwp357F8mIqg,1368
+atlasai_dstoolkit_client-0.0.9.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+atlasai_dstoolkit_client-0.0.9.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
+atlasai_dstoolkit_client-0.0.9.dist-info/RECORD,,