terrakio-core 0.4.98.1b15__py3-none-any.whl → 0.5.1__py3-none-any.whl

terrakio_core/__init__.py

@@ -8,7 +8,7 @@ Core components for Terrakio API clients.
 # Suppress ONNX Runtime GPU device discovery warnings - MUST BE FIRST!
 import os
 os.environ['ORT_LOGGING_LEVEL'] = '3'
-__version__ = "0.4.98.1"
+__version__ = "0.5.1"
 
 from .async_client import AsyncClient
 from .sync_client import SyncClient as Client

terrakio_core/endpoints/model_management.py

@@ -4,10 +4,6 @@ import textwrap
 from io import BytesIO
 from typing import Optional, Tuple
 
-import os
-
-os.environ['ORT_LOGGING_LEVEL'] = '3'
-
 import onnxruntime as ort
 
 from ..helper.decorators import require_api_key
@@ -919,13 +915,13 @@ class ModelManagement:
             if not SKL2ONNX_AVAILABLE:
                 raise ImportError("skl2onnx is not installed. Please install it with: pip install skl2onnx")
             return self._convert_sklearn_to_onnx(model, input_shape), "random_forest"
-        # elif isinstance(model, ort.InferenceSession):
-        #     if model_type is None:
-        #         raise ValueError(
-        #             "For ONNX InferenceSession models, you must specify the 'model_type' parameter. Currently 'nerual network' and 'random forest' are supported."
-        #             "Example: model_type='random forest' or model_type='neural network'"
-        #         )
-        #     return model.SerializeToString(), model_type
+        elif isinstance(model, ort.InferenceSession):
+            if model_type is None:
+                raise ValueError(
+                    "For ONNX InferenceSession models, you must specify the 'model_type' parameter. Currently 'nerual network' and 'random forest' are supported."
+                    "Example: model_type='random forest' or model_type='neural network'"
+                )
+            return model.SerializeToString(), model_type
         else:
             model_type = type(model).__name__
             raise ValueError(f"Unsupported model type: {model_type}. Supported types: PyTorch nn.Module, sklearn BaseEstimator")

{terrakio_core-0.4.98.1b15.dist-info → terrakio_core-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: terrakio-core
-Version: 0.4.98.1b15
+Version: 0.5.1
 Summary: Core package for the terrakio-python-api
 Requires-Python: >=3.11
 Requires-Dist: aiofiles>=24.1.0

{terrakio_core-0.4.98.1b15.dist-info → terrakio_core-0.5.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-terrakio_core/__init__.py,sha256=PfUM3-juKx4e5Qz2EG45lKEDUrplFjc3e42YkXWdm6k,394
+terrakio_core/__init__.py,sha256=qD9OAwIU1zNjl4tLvOoHW8ifquSiWQzLxeRiB9M-0mM,391
 terrakio_core/accessors.py,sha256=UZIi9y4RpBxouSmKuwuNYLIYqDxD8BH-GnUzwJuc1JI,47570
 terrakio_core/async_client.py,sha256=aCLjiAmhRS20efrA11hhGUbtwsxapIBp4hBAwG79K6Y,9766
 terrakio_core/client.py,sha256=VXP7BtJWIfpPPZR7_yNdSTcGwNgTwhb7KorusqkQrzk,5603
@@ -12,12 +12,12 @@ terrakio_core/endpoints/auth.py,sha256=5WvAO39aLsbJAVtxISwiOZseKr3B9I5tHjamPRehD
 terrakio_core/endpoints/dataset_management.py,sha256=jpwftiKOI59NhXXypqm6wtILIkfyjy9NfYifRIvhZS0,16791
 terrakio_core/endpoints/group_management.py,sha256=V0KOGTXwmePBFeym55G_m3ONR-jVj2IU4OVLiL5UKz4,15869
 terrakio_core/endpoints/mass_stats.py,sha256=KuUaqF9h2z__2UvrFwrVDVB3eXPGTvU8dHjy7NWll1g,42000
-terrakio_core/endpoints/model_management.py,sha256=_vdlG-xYrfcFfQvswS6t0t6LSn4eMmOnNm1mXvxGes0,51405
+terrakio_core/endpoints/model_management.py,sha256=tvJ4BOBsyXKKH430byGH25CkLIzXWgxaPaL0CvL8_0Y,51341
 terrakio_core/endpoints/space_management.py,sha256=YWb55nkJnFJGlALJ520DvurxDqVqwYtsvqQPWzxzhDs,2266
 terrakio_core/endpoints/user_management.py,sha256=L_g4ysrh2xyz_JbObUU_tCxgHxisrDUPntWgQOs15GE,7709
 terrakio_core/helper/bounded_taskgroup.py,sha256=wiTH10jhKZgrsgrFUNG6gig8bFkUEPHkGRT2XY7Rgmo,677
 terrakio_core/helper/decorators.py,sha256=L6om7wmWNgCei3Wy5U0aZ-70OzsCwclkjIf7SfQuhCg,2289
 terrakio_core/helper/tiles.py,sha256=lcLCO6KiP05lCI9vngo3zCZJ6Z9C0pUxHSQS4H58EHc,2699
-terrakio_core-0.4.98.1b15.dist-info/METADATA,sha256=NUVCbhCvvChBi-sNCHuoE24dQfy97bvnVi_eyHC0TP0,1185
-terrakio_core-0.4.98.1b15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-terrakio_core-0.4.98.1b15.dist-info/RECORD,,
+terrakio_core-0.5.1.dist-info/METADATA,sha256=BN9srGf8MFyrJs-JW57Ikpckr417r73ZtkGhypIDFGw,1179
+terrakio_core-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+terrakio_core-0.5.1.dist-info/RECORD,,