orca-sdk 0.0.97__py3-none-any.whl → 0.0.100__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 (299) hide show
  1. orca_sdk/__init__.py +1 -0
  2. orca_sdk/_shared/__init__.py +1 -0
  3. orca_sdk/_utils/analysis_ui.py +5 -5
  4. orca_sdk/_utils/auth.py +23 -33
  5. orca_sdk/_utils/pagination.py +126 -0
  6. orca_sdk/_utils/pagination_test.py +132 -0
  7. orca_sdk/classification_model.py +188 -126
  8. orca_sdk/classification_model_test.py +57 -8
  9. orca_sdk/client.py +3563 -0
  10. orca_sdk/conftest.py +10 -0
  11. orca_sdk/credentials.py +59 -21
  12. orca_sdk/credentials_test.py +20 -0
  13. orca_sdk/datasource.py +80 -93
  14. orca_sdk/datasource_test.py +41 -7
  15. orca_sdk/embedding_model.py +225 -71
  16. orca_sdk/embedding_model_test.py +27 -36
  17. orca_sdk/job.py +49 -45
  18. orca_sdk/job_test.py +16 -0
  19. orca_sdk/memoryset.py +340 -353
  20. orca_sdk/memoryset_test.py +7 -11
  21. orca_sdk/regression_model.py +120 -111
  22. orca_sdk/regression_model_test.py +15 -0
  23. orca_sdk/telemetry.py +162 -139
  24. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.100.dist-info}/METADATA +2 -5
  25. orca_sdk-0.0.100.dist-info/RECORD +40 -0
  26. orca_sdk/_generated_api_client/__init__.py +0 -3
  27. orca_sdk/_generated_api_client/api/__init__.py +0 -307
  28. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  29. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  30. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  31. orca_sdk/_generated_api_client/api/auth/create_org_plan_auth_org_plan_post.py +0 -168
  32. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  33. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  34. orca_sdk/_generated_api_client/api/auth/get_org_plan_auth_org_plan_get.py +0 -122
  35. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  36. orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +0 -168
  37. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  38. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +0 -170
  39. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py +0 -156
  40. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  41. orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py +0 -183
  42. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +0 -156
  43. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  44. orca_sdk/_generated_api_client/api/classification_model/list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  45. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/predict_label_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  47. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +0 -183
  48. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  49. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +0 -224
  50. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +0 -229
  51. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +0 -183
  52. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  53. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +0 -172
  54. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  55. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +0 -169
  56. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  57. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +0 -235
  58. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  59. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  60. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  61. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +0 -189
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  67. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  68. orca_sdk/_generated_api_client/api/memoryset/analyze_memoryset_memoryset_name_or_id_analysis_post.py +0 -183
  69. orca_sdk/_generated_api_client/api/memoryset/batch_delete_memoryset_batch_delete_memoryset_post.py +0 -168
  70. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  71. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  72. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  75. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  76. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -210
  77. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -186
  78. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  79. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -188
  80. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -235
  81. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -180
  82. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -212
  83. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +0 -195
  84. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -210
  85. orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py +0 -233
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -216
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -205
  88. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +0 -183
  89. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  90. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +0 -150
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  92. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -192
  93. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -161
  94. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  95. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  96. orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py +0 -170
  97. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  98. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py +0 -154
  99. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +0 -183
  100. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  101. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +0 -156
  102. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +0 -161
  103. orca_sdk/_generated_api_client/api/regression_model/list_regression_models_regression_model_get.py +0 -127
  104. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +0 -190
  105. orca_sdk/_generated_api_client/api/regression_model/update_regression_model_regression_model_name_or_id_patch.py +0 -183
  106. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  107. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  108. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  109. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +0 -156
  110. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -288
  111. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  112. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +0 -168
  113. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  114. orca_sdk/_generated_api_client/api/telemetry/explain_prediction_telemetry_prediction_prediction_id_explanation_get.py +0 -182
  115. orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +0 -239
  116. orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +0 -192
  117. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  118. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -180
  119. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  120. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +0 -198
  121. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -198
  122. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  123. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  124. orca_sdk/_generated_api_client/client.py +0 -216
  125. orca_sdk/_generated_api_client/errors.py +0 -38
  126. orca_sdk/_generated_api_client/models/__init__.py +0 -345
  127. orca_sdk/_generated_api_client/models/action_recommendation.py +0 -82
  128. orca_sdk/_generated_api_client/models/action_recommendation_action.py +0 -11
  129. orca_sdk/_generated_api_client/models/add_memory_recommendations.py +0 -85
  130. orca_sdk/_generated_api_client/models/add_memory_suggestion.py +0 -79
  131. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -116
  132. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -137
  133. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +0 -9
  134. orca_sdk/_generated_api_client/models/base_label_prediction_result.py +0 -130
  135. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  136. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +0 -108
  137. orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +0 -145
  138. orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py +0 -154
  139. orca_sdk/_generated_api_client/models/cascading_edit_suggestion.py +0 -92
  140. orca_sdk/_generated_api_client/models/class_representatives.py +0 -92
  141. orca_sdk/_generated_api_client/models/classification_evaluation_request.py +0 -148
  142. orca_sdk/_generated_api_client/models/classification_metrics.py +0 -259
  143. orca_sdk/_generated_api_client/models/classification_model_metadata.py +0 -227
  144. orca_sdk/_generated_api_client/models/classification_prediction_request.py +0 -220
  145. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +0 -210
  146. orca_sdk/_generated_api_client/models/cluster_metrics.py +0 -78
  147. orca_sdk/_generated_api_client/models/column_info.py +0 -145
  148. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  149. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +0 -81
  150. orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +0 -8
  151. orca_sdk/_generated_api_client/models/count_predictions_request.py +0 -195
  152. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -120
  153. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +0 -9
  154. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -145
  155. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +0 -9
  156. orca_sdk/_generated_api_client/models/create_classification_model_request.py +0 -237
  157. orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +0 -101
  158. orca_sdk/_generated_api_client/models/create_memoryset_request.py +0 -365
  159. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +0 -66
  160. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +0 -13
  161. orca_sdk/_generated_api_client/models/create_org_plan_request.py +0 -73
  162. orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +0 -11
  163. orca_sdk/_generated_api_client/models/create_regression_model_request.py +0 -157
  164. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -156
  165. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  166. orca_sdk/_generated_api_client/models/delete_memorysets_request.py +0 -70
  167. orca_sdk/_generated_api_client/models/embed_request.py +0 -155
  168. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +0 -205
  169. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +0 -197
  170. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +0 -158
  171. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +0 -86
  172. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  173. orca_sdk/_generated_api_client/models/embedding_model_result.py +0 -123
  174. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -153
  175. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +0 -140
  176. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +0 -140
  177. orca_sdk/_generated_api_client/models/feedback_metrics.py +0 -85
  178. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  179. orca_sdk/_generated_api_client/models/filter_item.py +0 -239
  180. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -17
  181. orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +0 -8
  182. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +0 -8
  183. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -22
  184. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  185. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  186. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  187. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  188. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  189. orca_sdk/_generated_api_client/models/http_validation_error.py +0 -86
  190. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -81
  191. orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +0 -8
  192. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  193. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -210
  194. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  195. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -288
  196. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -186
  197. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -128
  198. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  199. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -194
  200. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  201. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  202. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  203. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  204. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +0 -207
  205. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  206. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_metadata.py +0 -68
  207. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  208. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -319
  209. orca_sdk/_generated_api_client/models/lookup_request.py +0 -101
  210. orca_sdk/_generated_api_client/models/lookup_score_metrics.py +0 -94
  211. orca_sdk/_generated_api_client/models/memory_metrics.py +0 -263
  212. orca_sdk/_generated_api_client/models/memory_type.py +0 -9
  213. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -245
  214. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -105
  215. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -182
  216. orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +0 -79
  217. orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +0 -138
  218. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config.py +0 -202
  219. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_clustering_method.py +0 -9
  220. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_partitioning_method.py +0 -10
  221. orca_sdk/_generated_api_client/models/memoryset_cluster_metrics.py +0 -100
  222. orca_sdk/_generated_api_client/models/memoryset_duplicate_analysis_config.py +0 -70
  223. orca_sdk/_generated_api_client/models/memoryset_duplicate_metrics.py +0 -70
  224. orca_sdk/_generated_api_client/models/memoryset_label_analysis_config.py +0 -70
  225. orca_sdk/_generated_api_client/models/memoryset_label_metrics.py +0 -116
  226. orca_sdk/_generated_api_client/models/memoryset_metadata.py +0 -333
  227. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +0 -55
  228. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +0 -13
  229. orca_sdk/_generated_api_client/models/memoryset_metrics.py +0 -265
  230. orca_sdk/_generated_api_client/models/memoryset_neighbor_analysis_config.py +0 -83
  231. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics.py +0 -76
  232. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics_lookup_score_metrics.py +0 -68
  233. orca_sdk/_generated_api_client/models/memoryset_projection_analysis_config.py +0 -79
  234. orca_sdk/_generated_api_client/models/memoryset_projection_metrics.py +0 -55
  235. orca_sdk/_generated_api_client/models/memoryset_update.py +0 -121
  236. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -99
  237. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -23
  238. orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +0 -8
  239. orca_sdk/_generated_api_client/models/org_plan.py +0 -99
  240. orca_sdk/_generated_api_client/models/org_plan_tier.py +0 -11
  241. orca_sdk/_generated_api_client/models/paginated_task.py +0 -108
  242. orca_sdk/_generated_api_client/models/paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py +0 -135
  243. orca_sdk/_generated_api_client/models/pr_curve.py +0 -86
  244. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  245. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  246. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  247. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  248. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +0 -10
  249. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +0 -9
  250. orca_sdk/_generated_api_client/models/predictive_model_update.py +0 -111
  251. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -115
  252. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -17
  253. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  254. orca_sdk/_generated_api_client/models/rar_head_type.py +0 -8
  255. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +0 -148
  256. orca_sdk/_generated_api_client/models/regression_metrics.py +0 -172
  257. orca_sdk/_generated_api_client/models/regression_model_metadata.py +0 -191
  258. orca_sdk/_generated_api_client/models/regression_prediction_request.py +0 -195
  259. orca_sdk/_generated_api_client/models/roc_curve.py +0 -86
  260. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +0 -196
  261. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +0 -68
  262. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +0 -252
  263. orca_sdk/_generated_api_client/models/scored_memory.py +0 -172
  264. orca_sdk/_generated_api_client/models/scored_memory_insert.py +0 -128
  265. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +0 -68
  266. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +0 -180
  267. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +0 -68
  268. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +0 -68
  269. orca_sdk/_generated_api_client/models/scored_memory_update.py +0 -171
  270. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +0 -68
  271. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +0 -193
  272. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  273. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +0 -68
  274. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -81
  275. orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +0 -8
  276. orca_sdk/_generated_api_client/models/task.py +0 -198
  277. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  278. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  279. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +0 -8
  280. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_2.py +0 -9
  281. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +0 -8
  282. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +0 -8
  283. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +0 -217
  284. orca_sdk/_generated_api_client/models/telemetry_filter_item_op.py +0 -15
  285. orca_sdk/_generated_api_client/models/telemetry_memories_request.py +0 -181
  286. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +0 -185
  287. orca_sdk/_generated_api_client/models/telemetry_sort_options_direction.py +0 -9
  288. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -73
  289. orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +0 -8
  290. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -81
  291. orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +0 -8
  292. orca_sdk/_generated_api_client/models/update_org_plan_request.py +0 -73
  293. orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +0 -11
  294. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -133
  295. orca_sdk/_generated_api_client/models/validation_error.py +0 -99
  296. orca_sdk/_generated_api_client/py.typed +0 -1
  297. orca_sdk/_generated_api_client/types.py +0 -56
  298. orca_sdk-0.0.97.dist-info/RECORD +0 -309
  299. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.100.dist-info}/WHEEL +0 -0
@@ -1,148 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, List, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..types import UNSET, Unset
19
-
20
- T = TypeVar("T", bound="ClassificationEvaluationRequest")
21
-
22
-
23
- @_attrs_define
24
- class ClassificationEvaluationRequest:
25
- """
26
- Attributes:
27
- datasource_id (str):
28
- datasource_label_column (str):
29
- datasource_value_column (str):
30
- memoryset_override_id (Union[None, Unset, str]):
31
- record_telemetry (Union[Unset, bool]): Default: False.
32
- telemetry_tags (Union[List[str], None, Unset]):
33
- """
34
-
35
- datasource_id: str
36
- datasource_label_column: str
37
- datasource_value_column: str
38
- memoryset_override_id: Union[None, Unset, str] = UNSET
39
- record_telemetry: Union[Unset, bool] = False
40
- telemetry_tags: Union[List[str], None, Unset] = UNSET
41
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
42
-
43
- def to_dict(self) -> dict[str, Any]:
44
- datasource_id = self.datasource_id
45
-
46
- datasource_label_column = self.datasource_label_column
47
-
48
- datasource_value_column = self.datasource_value_column
49
-
50
- memoryset_override_id: Union[None, Unset, str]
51
- if isinstance(self.memoryset_override_id, Unset):
52
- memoryset_override_id = UNSET
53
- else:
54
- memoryset_override_id = self.memoryset_override_id
55
-
56
- record_telemetry = self.record_telemetry
57
-
58
- telemetry_tags: Union[List[str], None, Unset]
59
- if isinstance(self.telemetry_tags, Unset):
60
- telemetry_tags = UNSET
61
- elif isinstance(self.telemetry_tags, list):
62
- telemetry_tags = self.telemetry_tags
63
-
64
- else:
65
- telemetry_tags = self.telemetry_tags
66
-
67
- field_dict: dict[str, Any] = {}
68
- field_dict.update(self.additional_properties)
69
- field_dict.update(
70
- {
71
- "datasource_id": datasource_id,
72
- "datasource_label_column": datasource_label_column,
73
- "datasource_value_column": datasource_value_column,
74
- }
75
- )
76
- if memoryset_override_id is not UNSET:
77
- field_dict["memoryset_override_id"] = memoryset_override_id
78
- if record_telemetry is not UNSET:
79
- field_dict["record_telemetry"] = record_telemetry
80
- if telemetry_tags is not UNSET:
81
- field_dict["telemetry_tags"] = telemetry_tags
82
-
83
- return field_dict
84
-
85
- @classmethod
86
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
87
- d = src_dict.copy()
88
- datasource_id = d.pop("datasource_id")
89
-
90
- datasource_label_column = d.pop("datasource_label_column")
91
-
92
- datasource_value_column = d.pop("datasource_value_column")
93
-
94
- def _parse_memoryset_override_id(data: object) -> Union[None, Unset, str]:
95
- if data is None:
96
- return data
97
- if isinstance(data, Unset):
98
- return data
99
- return cast(Union[None, Unset, str], data)
100
-
101
- memoryset_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
102
-
103
- record_telemetry = d.pop("record_telemetry", UNSET)
104
-
105
- def _parse_telemetry_tags(data: object) -> Union[List[str], None, Unset]:
106
- if data is None:
107
- return data
108
- if isinstance(data, Unset):
109
- return data
110
- try:
111
- if not isinstance(data, list):
112
- raise TypeError()
113
- telemetry_tags_type_0 = cast(List[str], data)
114
-
115
- return telemetry_tags_type_0
116
- except: # noqa: E722
117
- pass
118
- return cast(Union[List[str], None, Unset], data)
119
-
120
- telemetry_tags = _parse_telemetry_tags(d.pop("telemetry_tags", UNSET))
121
-
122
- classification_evaluation_request = cls(
123
- datasource_id=datasource_id,
124
- datasource_label_column=datasource_label_column,
125
- datasource_value_column=datasource_value_column,
126
- memoryset_override_id=memoryset_override_id,
127
- record_telemetry=record_telemetry,
128
- telemetry_tags=telemetry_tags,
129
- )
130
-
131
- classification_evaluation_request.additional_properties = d
132
- return classification_evaluation_request
133
-
134
- @property
135
- def additional_keys(self) -> list[str]:
136
- return list(self.additional_properties.keys())
137
-
138
- def __getitem__(self, key: str) -> Any:
139
- return self.additional_properties[key]
140
-
141
- def __setitem__(self, key: str, value: Any) -> None:
142
- self.additional_properties[key] = value
143
-
144
- def __delitem__(self, key: str) -> None:
145
- del self.additional_properties[key]
146
-
147
- def __contains__(self, key: str) -> bool:
148
- return key in self.additional_properties
@@ -1,259 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..types import UNSET, Unset
19
-
20
- if TYPE_CHECKING:
21
- from ..models.pr_curve import PRCurve
22
- from ..models.roc_curve import ROCCurve
23
-
24
-
25
- T = TypeVar("T", bound="ClassificationMetrics")
26
-
27
-
28
- @_attrs_define
29
- class ClassificationMetrics:
30
- """
31
- Attributes:
32
- f1_score (float):
33
- accuracy (float):
34
- loss (float):
35
- anomaly_score_mean (Union[None, Unset, float]):
36
- anomaly_score_median (Union[None, Unset, float]):
37
- anomaly_score_variance (Union[None, Unset, float]):
38
- roc_auc (Union[None, Unset, float]):
39
- pr_auc (Union[None, Unset, float]):
40
- pr_curve (Union['PRCurve', None, Unset]):
41
- roc_curve (Union['ROCCurve', None, Unset]):
42
- """
43
-
44
- f1_score: float
45
- accuracy: float
46
- loss: float
47
- anomaly_score_mean: Union[None, Unset, float] = UNSET
48
- anomaly_score_median: Union[None, Unset, float] = UNSET
49
- anomaly_score_variance: Union[None, Unset, float] = UNSET
50
- roc_auc: Union[None, Unset, float] = UNSET
51
- pr_auc: Union[None, Unset, float] = UNSET
52
- pr_curve: Union["PRCurve", None, Unset] = UNSET
53
- roc_curve: Union["ROCCurve", None, Unset] = UNSET
54
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
55
-
56
- def to_dict(self) -> dict[str, Any]:
57
- from ..models.pr_curve import PRCurve
58
- from ..models.roc_curve import ROCCurve
59
-
60
- f1_score = self.f1_score
61
-
62
- accuracy = self.accuracy
63
-
64
- loss = self.loss
65
-
66
- anomaly_score_mean: Union[None, Unset, float]
67
- if isinstance(self.anomaly_score_mean, Unset):
68
- anomaly_score_mean = UNSET
69
- else:
70
- anomaly_score_mean = self.anomaly_score_mean
71
-
72
- anomaly_score_median: Union[None, Unset, float]
73
- if isinstance(self.anomaly_score_median, Unset):
74
- anomaly_score_median = UNSET
75
- else:
76
- anomaly_score_median = self.anomaly_score_median
77
-
78
- anomaly_score_variance: Union[None, Unset, float]
79
- if isinstance(self.anomaly_score_variance, Unset):
80
- anomaly_score_variance = UNSET
81
- else:
82
- anomaly_score_variance = self.anomaly_score_variance
83
-
84
- roc_auc: Union[None, Unset, float]
85
- if isinstance(self.roc_auc, Unset):
86
- roc_auc = UNSET
87
- else:
88
- roc_auc = self.roc_auc
89
-
90
- pr_auc: Union[None, Unset, float]
91
- if isinstance(self.pr_auc, Unset):
92
- pr_auc = UNSET
93
- else:
94
- pr_auc = self.pr_auc
95
-
96
- pr_curve: Union[Dict[str, Any], None, Unset]
97
- if isinstance(self.pr_curve, Unset):
98
- pr_curve = UNSET
99
- elif isinstance(self.pr_curve, PRCurve):
100
- pr_curve = self.pr_curve.to_dict()
101
- else:
102
- pr_curve = self.pr_curve
103
-
104
- roc_curve: Union[Dict[str, Any], None, Unset]
105
- if isinstance(self.roc_curve, Unset):
106
- roc_curve = UNSET
107
- elif isinstance(self.roc_curve, ROCCurve):
108
- roc_curve = self.roc_curve.to_dict()
109
- else:
110
- roc_curve = self.roc_curve
111
-
112
- field_dict: dict[str, Any] = {}
113
- field_dict.update(self.additional_properties)
114
- field_dict.update(
115
- {
116
- "f1_score": f1_score,
117
- "accuracy": accuracy,
118
- "loss": loss,
119
- }
120
- )
121
- if anomaly_score_mean is not UNSET:
122
- field_dict["anomaly_score_mean"] = anomaly_score_mean
123
- if anomaly_score_median is not UNSET:
124
- field_dict["anomaly_score_median"] = anomaly_score_median
125
- if anomaly_score_variance is not UNSET:
126
- field_dict["anomaly_score_variance"] = anomaly_score_variance
127
- if roc_auc is not UNSET:
128
- field_dict["roc_auc"] = roc_auc
129
- if pr_auc is not UNSET:
130
- field_dict["pr_auc"] = pr_auc
131
- if pr_curve is not UNSET:
132
- field_dict["pr_curve"] = pr_curve
133
- if roc_curve is not UNSET:
134
- field_dict["roc_curve"] = roc_curve
135
-
136
- return field_dict
137
-
138
- @classmethod
139
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
140
- from ..models.pr_curve import PRCurve
141
- from ..models.roc_curve import ROCCurve
142
-
143
- d = src_dict.copy()
144
- f1_score = d.pop("f1_score")
145
-
146
- accuracy = d.pop("accuracy")
147
-
148
- loss = d.pop("loss")
149
-
150
- def _parse_anomaly_score_mean(data: object) -> Union[None, Unset, float]:
151
- if data is None:
152
- return data
153
- if isinstance(data, Unset):
154
- return data
155
- return cast(Union[None, Unset, float], data)
156
-
157
- anomaly_score_mean = _parse_anomaly_score_mean(d.pop("anomaly_score_mean", UNSET))
158
-
159
- def _parse_anomaly_score_median(data: object) -> Union[None, Unset, float]:
160
- if data is None:
161
- return data
162
- if isinstance(data, Unset):
163
- return data
164
- return cast(Union[None, Unset, float], data)
165
-
166
- anomaly_score_median = _parse_anomaly_score_median(d.pop("anomaly_score_median", UNSET))
167
-
168
- def _parse_anomaly_score_variance(data: object) -> Union[None, Unset, float]:
169
- if data is None:
170
- return data
171
- if isinstance(data, Unset):
172
- return data
173
- return cast(Union[None, Unset, float], data)
174
-
175
- anomaly_score_variance = _parse_anomaly_score_variance(d.pop("anomaly_score_variance", UNSET))
176
-
177
- def _parse_roc_auc(data: object) -> Union[None, Unset, float]:
178
- if data is None:
179
- return data
180
- if isinstance(data, Unset):
181
- return data
182
- return cast(Union[None, Unset, float], data)
183
-
184
- roc_auc = _parse_roc_auc(d.pop("roc_auc", UNSET))
185
-
186
- def _parse_pr_auc(data: object) -> Union[None, Unset, float]:
187
- if data is None:
188
- return data
189
- if isinstance(data, Unset):
190
- return data
191
- return cast(Union[None, Unset, float], data)
192
-
193
- pr_auc = _parse_pr_auc(d.pop("pr_auc", UNSET))
194
-
195
- def _parse_pr_curve(data: object) -> Union["PRCurve", None, Unset]:
196
- if data is None:
197
- return data
198
- if isinstance(data, Unset):
199
- return data
200
- try:
201
- if not isinstance(data, dict):
202
- raise TypeError()
203
- pr_curve_type_0 = PRCurve.from_dict(data)
204
-
205
- return pr_curve_type_0
206
- except: # noqa: E722
207
- pass
208
- return cast(Union["PRCurve", None, Unset], data)
209
-
210
- pr_curve = _parse_pr_curve(d.pop("pr_curve", UNSET))
211
-
212
- def _parse_roc_curve(data: object) -> Union["ROCCurve", None, Unset]:
213
- if data is None:
214
- return data
215
- if isinstance(data, Unset):
216
- return data
217
- try:
218
- if not isinstance(data, dict):
219
- raise TypeError()
220
- roc_curve_type_0 = ROCCurve.from_dict(data)
221
-
222
- return roc_curve_type_0
223
- except: # noqa: E722
224
- pass
225
- return cast(Union["ROCCurve", None, Unset], data)
226
-
227
- roc_curve = _parse_roc_curve(d.pop("roc_curve", UNSET))
228
-
229
- classification_metrics = cls(
230
- f1_score=f1_score,
231
- accuracy=accuracy,
232
- loss=loss,
233
- anomaly_score_mean=anomaly_score_mean,
234
- anomaly_score_median=anomaly_score_median,
235
- anomaly_score_variance=anomaly_score_variance,
236
- roc_auc=roc_auc,
237
- pr_auc=pr_auc,
238
- pr_curve=pr_curve,
239
- roc_curve=roc_curve,
240
- )
241
-
242
- classification_metrics.additional_properties = d
243
- return classification_metrics
244
-
245
- @property
246
- def additional_keys(self) -> list[str]:
247
- return list(self.additional_properties.keys())
248
-
249
- def __getitem__(self, key: str) -> Any:
250
- return self.additional_properties[key]
251
-
252
- def __setitem__(self, key: str, value: Any) -> None:
253
- self.additional_properties[key] = value
254
-
255
- def __delitem__(self, key: str) -> None:
256
- del self.additional_properties[key]
257
-
258
- def __contains__(self, key: str) -> bool:
259
- return key in self.additional_properties
@@ -1,227 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import datetime
14
- from enum import Enum
15
- from typing import Any, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
- from dateutil.parser import isoparse
20
-
21
- from ..models.rac_head_type import RACHeadType
22
-
23
- T = TypeVar("T", bound="ClassificationModelMetadata")
24
-
25
-
26
- @_attrs_define
27
- class ClassificationModelMetadata:
28
- """
29
- Attributes:
30
- id (str):
31
- org_id (str):
32
- name (str):
33
- description (Union[None, str]):
34
- notes (Union[None, str]):
35
- version (int):
36
- memoryset_id (str):
37
- memory_lookup_count (int):
38
- storage_path (str):
39
- memoryset_collection_name (str):
40
- created_at (datetime.datetime):
41
- updated_at (datetime.datetime):
42
- locked (bool):
43
- num_classes (int):
44
- head_type (RACHeadType):
45
- weigh_memories (Union[None, bool]):
46
- min_memory_weight (Union[None, float]):
47
- """
48
-
49
- id: str
50
- org_id: str
51
- name: str
52
- description: Union[None, str]
53
- notes: Union[None, str]
54
- version: int
55
- memoryset_id: str
56
- memory_lookup_count: int
57
- storage_path: str
58
- memoryset_collection_name: str
59
- created_at: datetime.datetime
60
- updated_at: datetime.datetime
61
- locked: bool
62
- num_classes: int
63
- head_type: RACHeadType
64
- weigh_memories: Union[None, bool]
65
- min_memory_weight: Union[None, float]
66
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
67
-
68
- def to_dict(self) -> dict[str, Any]:
69
- id = self.id
70
-
71
- org_id = self.org_id
72
-
73
- name = self.name
74
-
75
- description: Union[None, str]
76
- description = self.description
77
-
78
- notes: Union[None, str]
79
- notes = self.notes
80
-
81
- version = self.version
82
-
83
- memoryset_id = self.memoryset_id
84
-
85
- memory_lookup_count = self.memory_lookup_count
86
-
87
- storage_path = self.storage_path
88
-
89
- memoryset_collection_name = self.memoryset_collection_name
90
-
91
- created_at = self.created_at.isoformat()
92
-
93
- updated_at = self.updated_at.isoformat()
94
-
95
- locked = self.locked
96
-
97
- num_classes = self.num_classes
98
-
99
- head_type = self.head_type.value if isinstance(self.head_type, Enum) else self.head_type
100
-
101
- weigh_memories: Union[None, bool]
102
- weigh_memories = self.weigh_memories
103
-
104
- min_memory_weight: Union[None, float]
105
- min_memory_weight = self.min_memory_weight
106
-
107
- field_dict: dict[str, Any] = {}
108
- field_dict.update(self.additional_properties)
109
- field_dict.update(
110
- {
111
- "id": id,
112
- "org_id": org_id,
113
- "name": name,
114
- "description": description,
115
- "notes": notes,
116
- "version": version,
117
- "memoryset_id": memoryset_id,
118
- "memory_lookup_count": memory_lookup_count,
119
- "storage_path": storage_path,
120
- "memoryset_collection_name": memoryset_collection_name,
121
- "created_at": created_at,
122
- "updated_at": updated_at,
123
- "locked": locked,
124
- "num_classes": num_classes,
125
- "head_type": head_type,
126
- "weigh_memories": weigh_memories,
127
- "min_memory_weight": min_memory_weight,
128
- }
129
- )
130
-
131
- return field_dict
132
-
133
- @classmethod
134
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
135
- d = src_dict.copy()
136
- id = d.pop("id")
137
-
138
- org_id = d.pop("org_id")
139
-
140
- name = d.pop("name")
141
-
142
- def _parse_description(data: object) -> Union[None, str]:
143
- if data is None:
144
- return data
145
- return cast(Union[None, str], data)
146
-
147
- description = _parse_description(d.pop("description"))
148
-
149
- def _parse_notes(data: object) -> Union[None, str]:
150
- if data is None:
151
- return data
152
- return cast(Union[None, str], data)
153
-
154
- notes = _parse_notes(d.pop("notes"))
155
-
156
- version = d.pop("version")
157
-
158
- memoryset_id = d.pop("memoryset_id")
159
-
160
- memory_lookup_count = d.pop("memory_lookup_count")
161
-
162
- storage_path = d.pop("storage_path")
163
-
164
- memoryset_collection_name = d.pop("memoryset_collection_name")
165
-
166
- created_at = isoparse(d.pop("created_at"))
167
-
168
- updated_at = isoparse(d.pop("updated_at"))
169
-
170
- locked = d.pop("locked")
171
-
172
- num_classes = d.pop("num_classes")
173
-
174
- head_type = RACHeadType(d.pop("head_type"))
175
-
176
- def _parse_weigh_memories(data: object) -> Union[None, bool]:
177
- if data is None:
178
- return data
179
- return cast(Union[None, bool], data)
180
-
181
- weigh_memories = _parse_weigh_memories(d.pop("weigh_memories"))
182
-
183
- def _parse_min_memory_weight(data: object) -> Union[None, float]:
184
- if data is None:
185
- return data
186
- return cast(Union[None, float], data)
187
-
188
- min_memory_weight = _parse_min_memory_weight(d.pop("min_memory_weight"))
189
-
190
- classification_model_metadata = cls(
191
- id=id,
192
- org_id=org_id,
193
- name=name,
194
- description=description,
195
- notes=notes,
196
- version=version,
197
- memoryset_id=memoryset_id,
198
- memory_lookup_count=memory_lookup_count,
199
- storage_path=storage_path,
200
- memoryset_collection_name=memoryset_collection_name,
201
- created_at=created_at,
202
- updated_at=updated_at,
203
- locked=locked,
204
- num_classes=num_classes,
205
- head_type=head_type,
206
- weigh_memories=weigh_memories,
207
- min_memory_weight=min_memory_weight,
208
- )
209
-
210
- classification_model_metadata.additional_properties = d
211
- return classification_model_metadata
212
-
213
- @property
214
- def additional_keys(self) -> list[str]:
215
- return list(self.additional_properties.keys())
216
-
217
- def __getitem__(self, key: str) -> Any:
218
- return self.additional_properties[key]
219
-
220
- def __setitem__(self, key: str, value: Any) -> None:
221
- self.additional_properties[key] = value
222
-
223
- def __delitem__(self, key: str) -> None:
224
- del self.additional_properties[key]
225
-
226
- def __contains__(self, key: str) -> bool:
227
- return key in self.additional_properties