orca-sdk 0.0.95__tar.gz → 0.0.96__tar.gz

{orca_sdk-0.0.95 → orca_sdk-0.0.96}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: orca_sdk
-Version: 0.0.95
+Version: 0.0.96
 Summary: SDK for interacting with Orca Services
 License: Apache-2.0
 Author: Orca DB Inc.

{orca_sdk-0.0.95 → orca_sdk-0.0.96}/orca_sdk/_generated_api_client/api/__init__.py

@@ -5,8 +5,8 @@ from .auth.create_api_key_auth_api_key_post import sync as create_api_key
 from .auth.delete_api_key_auth_api_key_name_or_id_delete import sync as delete_api_key
 from .auth.delete_org_auth_org_delete import sync as delete_org
 from .auth.list_api_keys_auth_api_key_get import sync as list_api_keys
-from .classification_model.create_classification_model_classification_model_post import (
-    sync as create_classification_model,
+from .classification_model.create_classification_model_gpu_classification_model_post import (
+    sync as create_classification_model_gpu,
 )
 from .classification_model.delete_classification_model_classification_model_name_or_id_delete import (
     sync as delete_classification_model,
@@ -139,8 +139,8 @@ from .pretrained_embedding_model.get_pretrained_embedding_model_pretrained_embed
 from .pretrained_embedding_model.list_pretrained_embedding_models_pretrained_embedding_model_get import (
     sync as list_pretrained_embedding_models,
 )
-from .regression_model.create_regression_model_regression_model_post import (
-    sync as create_regression_model,
+from .regression_model.create_regression_model_gpu_regression_model_post import (
+    sync as create_regression_model_gpu,
 )
 from .regression_model.delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete import (
     sync as delete_regression_model_evaluation,
@@ -212,12 +212,12 @@ __all__ = [
     "clone_memoryset",
     "count_predictions",
     "create_api_key",
-    "create_classification_model",
+    "create_classification_model_gpu",
     "create_datasource",
     "create_embedding_evaluation",
     "create_finetuned_embedding_model",
     "create_memoryset",
-    "create_regression_model",
+    "create_regression_model_gpu",
     "delete_api_key",
     "delete_classification_model",
     "delete_classification_model_evaluation",

orca_sdk-0.0.95/orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py → orca_sdk-0.0.96/orca_sdk/_generated_api_client/api/classification_model/create_classification_model_gpu_classification_model_post.py

@@ -30,7 +30,7 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": "/classification_model/",
+        "url": "/gpu/classification_model/",
     }
 
     _body = body.to_dict()

orca_sdk-0.0.95/orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py → orca_sdk-0.0.96/orca_sdk/_generated_api_client/api/regression_model/create_regression_model_gpu_regression_model_post.py

@@ -30,7 +30,7 @@ def _get_kwargs(
 
     _kwargs: dict[str, Any] = {
         "method": "post",
-        "url": "/regression_model/",
+        "url": "/gpu/regression_model/",
     }
 
     _body = body.to_dict()

{orca_sdk-0.0.95 → orca_sdk-0.0.96}/orca_sdk/classification_model.py

@@ -10,7 +10,7 @@ from uuid import UUID
 from datasets import Dataset
 
 from ._generated_api_client.api import (
-    create_classification_model,
+    create_classification_model_gpu,
     delete_classification_model,
     evaluate_classification_model,
     get_classification_model,
@@ -35,7 +35,10 @@ from ._generated_api_client.models import (
 from ._generated_api_client.models import (
     PredictionSortItemItemType1 as PredictionSortDirection,
 )
-from ._generated_api_client.models import PredictiveModelUpdate, RACHeadType
+from ._generated_api_client.models import (
+    PredictiveModelUpdate,
+    RACHeadType,
+)
 from ._generated_api_client.types import UNSET as CLIENT_UNSET
 from ._shared.metrics import ClassificationMetrics, calculate_classification_metrics
 from ._utils.common import UNSET, CreateMode, DropMode
@@ -203,7 +206,7 @@ class ClassificationModel:
 
                 return existing
 
-        metadata = create_classification_model(
+        metadata = create_classification_model_gpu(
             body=CreateClassificationModelRequest(
                 name=name,
                 memoryset_id=memoryset.id,

{orca_sdk-0.0.95 → orca_sdk-0.0.96}/orca_sdk/regression_model.py

@@ -11,7 +11,7 @@ import numpy as np
 from datasets import Dataset
 
 from ._generated_api_client.api import (
-    create_regression_model,
+    create_regression_model_gpu,
     delete_regression_model,
     evaluate_regression_model,
     get_regression_model,
@@ -178,7 +178,7 @@ class RegressionModel:
 
                 return existing
 
-        metadata = create_regression_model(
+        metadata = create_regression_model_gpu(
             body=CreateRegressionModelRequest(
                 name=name,
                 memoryset_id=memoryset.id,

{orca_sdk-0.0.95 → orca_sdk-0.0.96}/orca_sdk/regression_model_test.py

@@ -152,7 +152,6 @@ def test_evaluate(
     )
 
     assert isinstance(result, RegressionMetrics)
-    print(result, result.mse)
     assert np.allclose(result.mae, 0.4)
     assert 0.0 <= result.mse <= 1.0
     assert 0.0 <= result.rmse <= 1.0

{orca_sdk-0.0.95 → orca_sdk-0.0.96}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "orca_sdk"
-version = "0.0.95" # Will be set by CI before building the wheel from the git tag. Do not set manually.
+version = "0.0.96" # Will be set by CI before building the wheel from the git tag. Do not set manually.
 description = "SDK for interacting with Orca Services"
 authors = ["Orca DB Inc. <dev-rel@orcadb.ai>"]
 license = "Apache-2.0"