orca-sdk 0.0.95__py3-none-any.whl → 0.0.96__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (9) hide show
  1. orca_sdk/_generated_api_client/api/__init__.py +6 -6
  2. orca_sdk/_generated_api_client/api/classification_model/{create_classification_model_classification_model_post.py → create_classification_model_gpu_classification_model_post.py} +1 -1
  3. orca_sdk/_generated_api_client/api/regression_model/{create_regression_model_regression_model_post.py → create_regression_model_gpu_regression_model_post.py} +1 -1
  4. orca_sdk/classification_model.py +6 -3
  5. orca_sdk/regression_model.py +2 -2
  6. orca_sdk/regression_model_test.py +0 -1
  7. {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.96.dist-info}/METADATA +1 -1
  8. {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.96.dist-info}/RECORD +9 -9
  9. {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.96.dist-info}/WHEEL +0 -0
orca_sdk/_generated_api_client/api/__init__.py CHANGED
@@ -5,8 +5,8 @@ from .auth.create_api_key_auth_api_key_post import sync as create_api_key
5
5
  from .auth.delete_api_key_auth_api_key_name_or_id_delete import sync as delete_api_key
6
6
  from .auth.delete_org_auth_org_delete import sync as delete_org
7
7
  from .auth.list_api_keys_auth_api_key_get import sync as list_api_keys
8
- from .classification_model.create_classification_model_classification_model_post import (
9
- sync as create_classification_model,
8
+ from .classification_model.create_classification_model_gpu_classification_model_post import (
9
+ sync as create_classification_model_gpu,
10
10
  )
11
11
  from .classification_model.delete_classification_model_classification_model_name_or_id_delete import (
12
12
  sync as delete_classification_model,
@@ -139,8 +139,8 @@ from .pretrained_embedding_model.get_pretrained_embedding_model_pretrained_embed
139
139
  from .pretrained_embedding_model.list_pretrained_embedding_models_pretrained_embedding_model_get import (
140
140
  sync as list_pretrained_embedding_models,
141
141
  )
142
- from .regression_model.create_regression_model_regression_model_post import (
143
- sync as create_regression_model,
142
+ from .regression_model.create_regression_model_gpu_regression_model_post import (
143
+ sync as create_regression_model_gpu,
144
144
  )
145
145
  from .regression_model.delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete import (
146
146
  sync as delete_regression_model_evaluation,
@@ -212,12 +212,12 @@ __all__ = [
212
212
  "clone_memoryset",
213
213
  "count_predictions",
214
214
  "create_api_key",
215
- "create_classification_model",
215
+ "create_classification_model_gpu",
216
216
  "create_datasource",
217
217
  "create_embedding_evaluation",
218
218
  "create_finetuned_embedding_model",
219
219
  "create_memoryset",
220
- "create_regression_model",
220
+ "create_regression_model_gpu",
221
221
  "delete_api_key",
222
222
  "delete_classification_model",
223
223
  "delete_classification_model_evaluation",
orca_sdk/_generated_api_client/api/classification_model/{create_classification_model_classification_model_post.py → create_classification_model_gpu_classification_model_post.py} RENAMED
@@ -30,7 +30,7 @@ def _get_kwargs(
30
30
 
31
31
  _kwargs: dict[str, Any] = {
32
32
  "method": "post",
33
- "url": "/classification_model/",
33
+ "url": "/gpu/classification_model/",
34
34
  }
35
35
 
36
36
  _body = body.to_dict()
orca_sdk/_generated_api_client/api/regression_model/{create_regression_model_regression_model_post.py → create_regression_model_gpu_regression_model_post.py} RENAMED
@@ -30,7 +30,7 @@ def _get_kwargs(
30
30
 
31
31
  _kwargs: dict[str, Any] = {
32
32
  "method": "post",
33
- "url": "/regression_model/",
33
+ "url": "/gpu/regression_model/",
34
34
  }
35
35
 
36
36
  _body = body.to_dict()
orca_sdk/classification_model.py CHANGED
@@ -10,7 +10,7 @@ from uuid import UUID
10
10
  from datasets import Dataset
11
11
 
12
12
  from ._generated_api_client.api import (
13
- create_classification_model,
13
+ create_classification_model_gpu,
14
14
  delete_classification_model,
15
15
  evaluate_classification_model,
16
16
  get_classification_model,
@@ -35,7 +35,10 @@ from ._generated_api_client.models import (
35
35
  from ._generated_api_client.models import (
36
36
  PredictionSortItemItemType1 as PredictionSortDirection,
37
37
  )
38
- from ._generated_api_client.models import PredictiveModelUpdate, RACHeadType
38
+ from ._generated_api_client.models import (
39
+ PredictiveModelUpdate,
40
+ RACHeadType,
41
+ )
39
42
  from ._generated_api_client.types import UNSET as CLIENT_UNSET
40
43
  from ._shared.metrics import ClassificationMetrics, calculate_classification_metrics
41
44
  from ._utils.common import UNSET, CreateMode, DropMode
@@ -203,7 +206,7 @@ class ClassificationModel:
203
206
 
204
207
  return existing
205
208
 
206
- metadata = create_classification_model(
209
+ metadata = create_classification_model_gpu(
207
210
  body=CreateClassificationModelRequest(
208
211
  name=name,
209
212
  memoryset_id=memoryset.id,
orca_sdk/regression_model.py CHANGED
@@ -11,7 +11,7 @@ import numpy as np
11
11
  from datasets import Dataset
12
12
 
13
13
  from ._generated_api_client.api import (
14
- create_regression_model,
14
+ create_regression_model_gpu,
15
15
  delete_regression_model,
16
16
  evaluate_regression_model,
17
17
  get_regression_model,
@@ -178,7 +178,7 @@ class RegressionModel:
178
178
 
179
179
  return existing
180
180
 
181
- metadata = create_regression_model(
181
+ metadata = create_regression_model_gpu(
182
182
  body=CreateRegressionModelRequest(
183
183
  name=name,
184
184
  memoryset_id=memoryset.id,
orca_sdk/regression_model_test.py CHANGED
@@ -152,7 +152,6 @@ def test_evaluate(
152
152
  )
153
153
 
154
154
  assert isinstance(result, RegressionMetrics)
155
- print(result, result.mse)
156
155
  assert np.allclose(result.mae, 0.4)
157
156
  assert 0.0 <= result.mse <= 1.0
158
157
  assert 0.0 <= result.rmse <= 1.0
{orca_sdk-0.0.95.dist-info → orca_sdk-0.0.96.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: orca_sdk
3
- Version: 0.0.95
3
+ Version: 0.0.96
4
4
  Summary: SDK for interacting with Orca Services
5
5
  License: Apache-2.0
6
6
  Author: Orca DB Inc.
{orca_sdk-0.0.95.dist-info → orca_sdk-0.0.96.dist-info}/RECORD RENAMED
@@ -1,6 +1,6 @@
1
1
  orca_sdk/__init__.py,sha256=I8bJHN2L1zB6nU5buMNVvDNNSiElmk47-jiDhaO0-Uk,989
2
2
  orca_sdk/_generated_api_client/__init__.py,sha256=2-BPUe-gQ4yMDdxcfBZ1PEHKn-te41oFdKgrzK29JV0,69
3
- orca_sdk/_generated_api_client/api/__init__.py,sha256=qA_ZlS7n32qABolKfHreYlhelv3tBB4zdrdKtmtvmkU,11896
3
+ orca_sdk/_generated_api_client/api/__init__.py,sha256=V9hywdinmbeLgsluO2AmoDTh7gaHVloqCMYAo9ZF7iY,11920
4
4
  orca_sdk/_generated_api_client/api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py,sha256=lARAAY0SCTCBUPRo9r31yGoJZ2RklwaqD6kQBnsXfsY,3900
6
6
  orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py,sha256=HeSjn9M8vplZdoeZE8jYyHq_s_chx4ElwF6FZptI_f4,4570
@@ -8,7 +8,7 @@ orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_d
8
8
  orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py,sha256=xan8cbfkiq81un8VVT1LOsw49kOn6GJIWK6PCdRzfR4,3741
9
9
  orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py,sha256=a-6ZspVGbHQTnM0urNiBqCOcKBQCgZpV4yhCMmO2PnU,3931
10
10
  orca_sdk/_generated_api_client/api/classification_model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
- orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py,sha256=lva5w7DPfKJq_A0G7_C_E_9JKvX6NY6TC8c-rtXOOyw,4864
11
+ orca_sdk/_generated_api_client/api/classification_model/create_classification_model_gpu_classification_model_post.py,sha256=q0H-Y3OCDd4v-C-PfHGlWBZ-aIIX4vqjrOGeaOwtDYk,4868
12
12
  orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py,sha256=V8EotTnaiVNQOiznKUHgBbJmzxVqwZVpFC_lR2Vny40,4137
13
13
  orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py,sha256=-2gAZQf6gA26-iHvyZjjwvnymSA0ih4E0nVx_ia8Jt0,4544
14
14
  orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py,sha256=z9j4ZV8ptFZrrxAiTUhQQvmKnfjl7pc_q3xQ01ZreNU,5211
@@ -65,7 +65,7 @@ orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrai
65
65
  orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py,sha256=k5nJ3FeiQHxjsoUBIvlHmZkIuOX5wONjSQZ3ZtjoX9s,5128
66
66
  orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py,sha256=3vfPcJ-MyLC93g_N8cOYf44zbJ1VQ7ldYE2Kv6-cIe0,4257
67
67
  orca_sdk/_generated_api_client/api/regression_model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
68
- orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py,sha256=fU4-Y-D-ihQTqO-sBB5TCRzOup3q6gTRZCT_jWRLq6w,4748
68
+ orca_sdk/_generated_api_client/api/regression_model/create_regression_model_gpu_regression_model_post.py,sha256=WSluoKXAHdtIO6zd4yTvR0qws7kzUFcV3hPZUqDMoSI,4752
69
69
  orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete.py,sha256=o-ZI3NHXT33zKw4x0N82AbDBm8oi5Ox044cnplRV4yw,4524
70
70
  orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py,sha256=yJCTt-bI5NFaWk3mFA8hd7zZ6rnUf7OSLk49-RWi0ks,4034
71
71
  orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py,sha256=7v4uqGsYrV-lnWor3rLlJ192S5x7RJi4BI1R68hXD0E,5147
@@ -256,7 +256,7 @@ orca_sdk/_utils/prediction_result_ui.py,sha256=WxsIHVLvJW0yE6QrBQHWn0BjPUFcDYNF3
256
256
  orca_sdk/_utils/tqdm_file_reader.py,sha256=Lw7Cg1UgNuRUoN6jjqZb-IlV00H-kbRcrZLdudr1GxE,324
257
257
  orca_sdk/_utils/value_parser.py,sha256=c3qMABCCDQcIjn9N1orYYnlRwDW9JWdGwW_2TDZPLdI,1286
258
258
  orca_sdk/_utils/value_parser_test.py,sha256=OybsiC-Obi32RRi9NIuwrVBRAnlyPMV1xVAaevSrb7M,1079
259
- orca_sdk/classification_model.py,sha256=Pt7dBfHmHWXQvXuac89cXiC-1TefEJi2irmCstGCWHI,28279
259
+ orca_sdk/classification_model.py,sha256=TKvDx3DOtg-uzXlTlAfiOxdogJA4DH_-WwUAPayfS3Q,28300
260
260
  orca_sdk/classification_model_test.py,sha256=wxKdvuBjaalC-O1SLiTjc93TkDqWILEhQ2v5p4WM0TI,15734
261
261
  orca_sdk/conftest.py,sha256=pjErmIcjEzu0WgNEBTU52ytrGO1eaPQ648cfo26qByw,8687
262
262
  orca_sdk/credentials.py,sha256=gq_4w_o-igCCLNR6TY1x4RzMYysKUCsXJvdi6nem-A0,3558
@@ -269,10 +269,10 @@ orca_sdk/job.py,sha256=ER9SMvIvJtN-Vl4aHNn7L-qJH8RugSPZuOrrIDjntCw,12366
269
269
  orca_sdk/job_test.py,sha256=VLEuopck5oc_7Cf-D69tF5eUZzn9749Woh5rDA7LElw,1907
270
270
  orca_sdk/memoryset.py,sha256=r2TyDxnaoMCbvsaYPpgqltdz4bVtWA5B0h8TWGMC2l8,82895
271
271
  orca_sdk/memoryset_test.py,sha256=loSi6IWs8FNgn23w-lV_5z77wB8T4SIpiSI2OkPpQeI,20708
272
- orca_sdk/regression_model.py,sha256=gSrUS2hO-Dni5TXSxClZ9lF0pgxkf7mD52VuUtUAexY,23975
273
- orca_sdk/regression_model_test.py,sha256=1i9J-M-kJ-7qqLSBjpGvKwxNkWSUgJ2q1JV5jlWq4MA,13876
272
+ orca_sdk/regression_model.py,sha256=3PP9ue8r0Kgw4iVFPTzOfRu9e7zvZ7AXjXEad7L7K60,23983
273
+ orca_sdk/regression_model_test.py,sha256=69aORvjyB_Xuiu4CxeEIZQPwRj4uNZtdhbiTNdp3POU,13846
274
274
  orca_sdk/telemetry.py,sha256=yzGNfERdvhRAu9f0up36pRxXqVFBMC8rqklbTPrO8DA,21181
275
275
  orca_sdk/telemetry_test.py,sha256=eT66C5lFdNg-pQdo2I__BP7Tn5fTc9aTkVo9ZhWwhU0,5519
276
- orca_sdk-0.0.95.dist-info/METADATA,sha256=A6EYFtSiKx7nWaY69wBFKhylGYsS5WnsG-M_yeEj9iY,3146
277
- orca_sdk-0.0.95.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
278
- orca_sdk-0.0.95.dist-info/RECORD,,
276
+ orca_sdk-0.0.96.dist-info/METADATA,sha256=tsIvrFZxH6gvBcKdfiTyZgPZY2w4KVaz8XDUnCyod60,3146
277
+ orca_sdk-0.0.96.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
278
+ orca_sdk-0.0.96.dist-info/RECORD,,