terrakio-core 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

terrakio_core/client.py

@@ -1,12 +1,16 @@
-import requests
-import xarray as xr
+import json
+import asyncio
 from io import BytesIO
 from typing import Dict, Any, Optional, Union
-import json
+
+import requests
 import aiohttp
-import asyncio
-from shapely.geometry import shape
+import pandas as pd
+import geopandas as gpd
+import xarray as xr
+from shapely.geometry import shape, mapping
 from shapely.geometry.base import BaseGeometry as ShapelyGeometry
+
 from .exceptions import APIError, ConfigurationError
 
 class BaseClient:
@@ -671,10 +675,6 @@ class BaseClient:
         """
         Compute zonal statistics for all geometries in a GeoDataFrame using asyncio for concurrency.
         """
-        import asyncio
-        import pandas as pd
-        import geopandas as gpd
-        from shapely.geometry import mapping
         
         # Process geometries in batches
         all_results = []
@@ -971,3 +971,35 @@ class BaseClient:
             )
         return self.space_management.delete_data_in_path(path, region)
 
+    def train_model(self, model_name: str, training_data: dict) -> dict:
+        """
+        Train a model using the external model training API.
+
+        Args:
+            model_name (str): The name of the model to train.
+            training_data (dict): Dictionary containing training data parameters.
+
+        Returns:
+            dict: The response from the model training API.
+        """
+        endpoint = "https://modeltraining-573248941006.australia-southeast1.run.app/train_model"
+        payload = {
+            "model_name": model_name,
+            "training_data": training_data
+        }
+        try:
+            response = self.session.post(endpoint, json=payload, timeout=self.timeout, verify=self.verify)
+            if not response.ok:
+                error_msg = f"Model training request failed: {response.status_code} {response.reason}"
+                try:
+                    error_data = response.json()
+                    if "detail" in error_data:
+                        error_msg += f" - {error_data['detail']}"
+                except Exception:
+                    if response.text:
+                        error_msg += f" - {response.text}"
+                raise APIError(error_msg)
+            return response.json()
+        except requests.RequestException as e:
+            raise APIError(f"Model training request failed: {str(e)}")
+

{terrakio_core-0.2.2.dist-info → terrakio_core-0.2.4.dist-info}/METADATA

@@ -1,9 +1,7 @@
 Metadata-Version: 2.4
 Name: terrakio-core
-Version: 0.2.2
+Version: 0.2.4
 Summary: Core components for Terrakio API clients
-Home-page: https://github.com/HaizeaAnalytics/terrakio-python-api
-Author: Yupeng Chao
 Author-email: Yupeng Chao <yupeng@haizea.com.au>
 Project-URL: Homepage, https://github.com/HaizeaAnalytics/terrakio-python-api
 Project-URL: Bug Tracker, https://github.com/HaizeaAnalytics/terrakio-python-api/issues
@@ -15,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Development Status :: 4 - Beta
-Requires-Python: >=3.7
+Requires-Python: >3.11
 Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.25.0
 Requires-Dist: aiohttp>=3.8.0
@@ -23,9 +21,6 @@ Requires-Dist: pyyaml>=5.1
 Requires-Dist: xarray>=2023.1.0
 Requires-Dist: shapely>=2.0.0
 Requires-Dist: geopandas>=0.13.0
-Dynamic: author
-Dynamic: home-page
-Dynamic: requires-python
 
 # Terrakio Core
 

{terrakio_core-0.2.2.dist-info → terrakio_core-0.2.4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 terrakio_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 terrakio_core/auth.py,sha256=Nuj0_X3Hiy17svYgGxrSAR-LXpTlP0J0dSrfMnkPUbI,7717
-terrakio_core/client.py,sha256=uxu11vA_G4uZP1TaE0f9C1zjl4bh8oswBbYst-Jb4CU,43555
+terrakio_core/client.py,sha256=ChKbsGsCbkngRZ8mWgAijhKzElL0ZSdRHU2_XF9C7Ss,44850
 terrakio_core/config.py,sha256=AwJ1VgR5K7N32XCU5k7_Dp1nIv_FYt8MBonq9yKlGzA,2658
 terrakio_core/dataset_management.py,sha256=hhO35fwStS6HYFQdKP9wkr3DxHgjvpctmIU8UWH6w6U,8742
 terrakio_core/exceptions.py,sha256=9S-I20-QiDRj1qgjFyYUwYM7BLic_bxurcDOIm2Fu_0,410
@@ -8,7 +8,7 @@ terrakio_core/group_access_management.py,sha256=NJ7SX4keUzZAUENmJ5L6ynKf4eRlqtyi
 terrakio_core/mass_stats.py,sha256=AqYJsd6nqo2BDh4vEPUDgsv4T0UR1_TPDoXa3WO3gTU,9284
 terrakio_core/space_management.py,sha256=wlUUQrlj_4U_Lpjn9lbF5oj0Rv3NPvvnrd5mWej5kmA,4211
 terrakio_core/user_management.py,sha256=MMNWkz0V_9X7ZYjjteuRU4H4W3F16iuQw1dpA2wVTGg,7400
-terrakio_core-0.2.2.dist-info/METADATA,sha256=hJV6bpoJSybkz_6J4oCXvGeH5HnVfGL5qpfQkYQ0KjI,1551
-terrakio_core-0.2.2.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-terrakio_core-0.2.2.dist-info/top_level.txt,sha256=5cBj6O7rNWyn97ND4YuvvXm0Crv4RxttT4JZvNdOG6Q,14
-terrakio_core-0.2.2.dist-info/RECORD,,
+terrakio_core-0.2.4.dist-info/METADATA,sha256=dMdHIPIFLVF1UMfTSa7D3GMLOw6nVJwl4hMeMcs0FdQ,1405
+terrakio_core-0.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+terrakio_core-0.2.4.dist-info/top_level.txt,sha256=5cBj6O7rNWyn97ND4YuvvXm0Crv4RxttT4JZvNdOG6Q,14
+terrakio_core-0.2.4.dist-info/RECORD,,

{terrakio_core-0.2.2.dist-info → terrakio_core-0.2.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any