truefoundry 0.5.10rc1__py3-none-any.whl → 0.5.11rc1__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.

Potentially problematic release.


This version of truefoundry might be problematic. Click here for more details.

Files changed (266) hide show
  1. truefoundry/__init__.py +19 -0
  2. truefoundry/_client.py +36 -0
  3. truefoundry/cli/display_util.py +1 -1
  4. truefoundry/common/utils.py +9 -0
  5. truefoundry/deploy/__init__.py +2 -2
  6. truefoundry/deploy/_autogen/models.py +1591 -0
  7. truefoundry/deploy/builder/__init__.py +1 -1
  8. truefoundry/deploy/builder/builders/dockerfile.py +1 -1
  9. truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py +1 -1
  10. truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py +1 -1
  11. truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py +1 -1
  12. truefoundry/deploy/io/output_callback.py +13 -6
  13. truefoundry/deploy/io/rich_output_callback.py +8 -4
  14. truefoundry/deploy/lib/clients/servicefoundry_client.py +117 -138
  15. truefoundry/deploy/lib/dao/apply.py +2 -2
  16. truefoundry/deploy/lib/dao/delete.py +2 -2
  17. truefoundry/deploy/lib/model/entity.py +37 -24
  18. truefoundry/deploy/v2/lib/deploy.py +7 -7
  19. truefoundry/deploy/v2/lib/deploy_workflow.py +9 -9
  20. truefoundry/deploy/v2/lib/deployable_patched_models.py +1 -1
  21. truefoundry/deploy/v2/lib/patched_models.py +7 -7
  22. truefoundry/deploy/v2/lib/source.py +1 -1
  23. truefoundry/ml/__init__.py +2 -2
  24. truefoundry/ml/_autogen/client/__init__.py +457 -0
  25. truefoundry/ml/_autogen/client/api/__init__.py +16 -0
  26. truefoundry/ml/{autogen → _autogen}/client/api/auth_api.py +6 -6
  27. truefoundry/ml/{autogen → _autogen}/client/api/deprecated_api.py +12 -345
  28. truefoundry/ml/{autogen → _autogen}/client/api/experiments_api.py +89 -44
  29. truefoundry/ml/{autogen → _autogen}/client/api/generate_code_snippet_api.py +10 -10
  30. truefoundry/ml/{autogen → _autogen}/client/api/health_api.py +5 -5
  31. truefoundry/ml/{autogen → _autogen}/client/api/metrics_api.py +10 -10
  32. truefoundry/ml/{autogen → _autogen}/client/api/mlfoundry_artifacts_api.py +589 -197
  33. truefoundry/ml/{autogen → _autogen}/client/api/run_artifacts_api.py +14 -9
  34. truefoundry/ml/{autogen → _autogen}/client/api/runs_api.py +41 -41
  35. truefoundry/ml/{autogen → _autogen}/client/api_client.py +8 -8
  36. truefoundry/ml/{autogen → _autogen}/client/configuration.py +30 -58
  37. truefoundry/ml/{autogen → _autogen}/client/exceptions.py +2 -2
  38. truefoundry/ml/_autogen/client/models/__init__.py +428 -0
  39. truefoundry/ml/{autogen → _autogen}/client/models/agent_app.py +14 -7
  40. truefoundry/ml/{autogen/client/models/agent.py → _autogen/client/models/agent_manifest.py} +37 -35
  41. truefoundry/ml/{autogen/client/models/agent_open_api_tool.py → _autogen/client/models/agent_open_api_tool_manifest.py} +50 -37
  42. truefoundry/ml/{autogen → _autogen}/client/models/agent_open_api_tool_with_fqn.py +45 -32
  43. truefoundry/ml/{autogen → _autogen}/client/models/agent_with_fqn.py +32 -30
  44. truefoundry/ml/_autogen/client/models/apply_request_dto.py +75 -0
  45. truefoundry/ml/_autogen/client/models/apply_response_dto.py +79 -0
  46. truefoundry/ml/{autogen → _autogen}/client/models/artifact_dto.py +7 -5
  47. truefoundry/ml/{autogen/client/models/artifact_version_manifest.py → _autogen/client/models/artifact_manifest.py} +39 -31
  48. truefoundry/ml/_autogen/client/models/artifact_path.py +70 -0
  49. truefoundry/ml/{autogen → _autogen}/client/models/artifact_response_dto.py +3 -3
  50. truefoundry/ml/{autogen → _autogen}/client/models/artifact_type.py +3 -3
  51. truefoundry/ml/{autogen → _autogen}/client/models/artifact_version_dto.py +10 -18
  52. truefoundry/ml/{autogen → _autogen}/client/models/artifact_version_response_dto.py +5 -3
  53. truefoundry/ml/{autogen → _autogen}/client/models/artifact_version_serialization_format.py +2 -2
  54. truefoundry/ml/{autogen → _autogen}/client/models/artifact_version_status.py +2 -2
  55. truefoundry/ml/{autogen → _autogen}/client/models/assistant_message.py +6 -6
  56. truefoundry/ml/{autogen → _autogen}/client/models/authorize_user_for_model_request_dto.py +2 -2
  57. truefoundry/ml/{autogen → _autogen}/client/models/authorize_user_for_model_version_request_dto.py +2 -2
  58. truefoundry/ml/{autogen → _autogen}/client/models/blob_storage_reference.py +6 -9
  59. truefoundry/ml/{autogen → _autogen}/client/models/body_get_search_runs_get.py +2 -2
  60. truefoundry/ml/{autogen/client/models/chat_prompt.py → _autogen/client/models/chat_prompt_manifest.py} +39 -37
  61. truefoundry/ml/{autogen/client/models/chat_prompt_messages_inner.py → _autogen/client/models/chat_prompt_manifest_messages_inner.py} +15 -15
  62. truefoundry/ml/{autogen → _autogen}/client/models/columns_dto.py +2 -2
  63. truefoundry/ml/{autogen → _autogen}/client/models/command.py +2 -2
  64. truefoundry/ml/{autogen → _autogen}/client/models/content.py +4 -4
  65. truefoundry/ml/{autogen → _autogen}/client/models/content1.py +4 -4
  66. truefoundry/ml/{autogen → _autogen}/client/models/content2.py +5 -5
  67. truefoundry/ml/{autogen → _autogen}/client/models/content2_any_of_inner.py +4 -4
  68. truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_request_dto.py +3 -3
  69. truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_response_dto.py +2 -2
  70. truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_version_request_dto.py +3 -3
  71. truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_version_response_dto.py +2 -2
  72. truefoundry/ml/{autogen → _autogen}/client/models/create_dataset_request_dto.py +21 -5
  73. truefoundry/ml/{autogen → _autogen}/client/models/create_experiment_request_dto.py +3 -3
  74. truefoundry/ml/{autogen → _autogen}/client/models/create_experiment_response_dto.py +2 -2
  75. truefoundry/ml/{autogen → _autogen}/client/models/create_model_version_request_dto.py +4 -12
  76. truefoundry/ml/{autogen → _autogen}/client/models/create_multi_part_upload_for_dataset_request_dto.py +2 -2
  77. truefoundry/ml/{autogen → _autogen}/client/models/create_multi_part_upload_for_dataset_response_dto.py +3 -3
  78. truefoundry/ml/{autogen → _autogen}/client/models/create_multi_part_upload_request_dto.py +2 -2
  79. truefoundry/ml/{autogen → _autogen}/client/models/create_python_deployment_config_request_dto.py +2 -2
  80. truefoundry/ml/{autogen → _autogen}/client/models/create_python_deployment_config_response_dto.py +2 -2
  81. truefoundry/ml/{autogen → _autogen}/client/models/create_run_request_dto.py +3 -3
  82. truefoundry/ml/{autogen → _autogen}/client/models/create_run_response_dto.py +3 -3
  83. truefoundry/ml/{autogen → _autogen}/client/models/create_workflow_task_config_request_dto.py +2 -2
  84. truefoundry/ml/_autogen/client/models/data_directory_manifest.py +124 -0
  85. truefoundry/ml/{autogen → _autogen}/client/models/dataset_dto.py +15 -4
  86. truefoundry/ml/{autogen → _autogen}/client/models/dataset_response_dto.py +3 -3
  87. truefoundry/ml/{autogen → _autogen}/client/models/delete_artifact_versions_request_dto.py +2 -2
  88. truefoundry/ml/{autogen → _autogen}/client/models/delete_dataset_request_dto.py +2 -2
  89. truefoundry/ml/{autogen → _autogen}/client/models/delete_files_for_dataset_request_dto.py +2 -2
  90. truefoundry/ml/{autogen → _autogen}/client/models/delete_model_version_request_dto.py +2 -2
  91. truefoundry/ml/{autogen → _autogen}/client/models/delete_run_request.py +2 -2
  92. truefoundry/ml/{autogen → _autogen}/client/models/delete_tag_request_dto.py +2 -2
  93. truefoundry/ml/{autogen → _autogen}/client/models/experiment_dto.py +5 -5
  94. truefoundry/ml/{autogen → _autogen}/client/models/experiment_id_request_dto.py +2 -2
  95. truefoundry/ml/{autogen → _autogen}/client/models/experiment_response_dto.py +3 -3
  96. truefoundry/ml/{autogen → _autogen}/client/models/experiment_tag_dto.py +2 -2
  97. truefoundry/ml/{autogen → _autogen}/client/models/export_deployment_files_request_dto.py +3 -3
  98. truefoundry/ml/{autogen → _autogen}/client/models/external_blob_storage_source.py +17 -9
  99. truefoundry/ml/{autogen → _autogen}/client/models/fast_ai_framework.py +4 -7
  100. truefoundry/ml/{autogen → _autogen}/client/models/file_info_dto.py +6 -3
  101. truefoundry/ml/{autogen → _autogen}/client/models/finalize_artifact_version_request_dto.py +7 -15
  102. truefoundry/ml/{autogen → _autogen}/client/models/framework.py +17 -17
  103. truefoundry/ml/{autogen → _autogen}/client/models/get_artifact_version_aliases_response_dto.py +2 -2
  104. truefoundry/ml/{autogen → _autogen}/client/models/get_experiment_response_dto.py +4 -4
  105. truefoundry/ml/{autogen → _autogen}/client/models/get_latest_run_log_response_dto.py +3 -3
  106. truefoundry/ml/{autogen → _autogen}/client/models/get_metric_history_response.py +3 -3
  107. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_url_for_dataset_write_request_dto.py +2 -2
  108. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_artifact_version_read_request_dto.py +2 -2
  109. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_artifact_version_read_response_dto.py +3 -3
  110. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_artifact_version_write_request_dto.py +2 -2
  111. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_artifact_version_write_response_dto.py +3 -3
  112. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_dataset_read_request_dto.py +2 -2
  113. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_dataset_read_response_dto.py +3 -3
  114. truefoundry/ml/{autogen → _autogen}/client/models/get_signed_urls_for_dataset_write_response_dto.py +3 -3
  115. truefoundry/ml/{autogen → _autogen}/client/models/get_tenant_id_response_dto.py +2 -2
  116. truefoundry/ml/{autogen → _autogen}/client/models/gluon_framework.py +4 -6
  117. truefoundry/ml/{autogen → _autogen}/client/models/h2_o_framework.py +4 -6
  118. truefoundry/ml/{autogen → _autogen}/client/models/http_validation_error.py +3 -3
  119. truefoundry/ml/{autogen → _autogen}/client/models/image_content_part.py +6 -7
  120. truefoundry/ml/{autogen → _autogen}/client/models/image_url.py +4 -4
  121. truefoundry/ml/{autogen → _autogen}/client/models/infer_method_name.py +3 -3
  122. truefoundry/ml/{autogen → _autogen}/client/models/keras_framework.py +4 -6
  123. truefoundry/ml/{autogen → _autogen}/client/models/latest_run_log_dto.py +3 -3
  124. truefoundry/ml/{autogen → _autogen}/client/models/library_name.py +3 -3
  125. truefoundry/ml/{autogen → _autogen}/client/models/light_gbm_framework.py +4 -7
  126. truefoundry/ml/{autogen → _autogen}/client/models/list_artifact_versions_request_dto.py +4 -4
  127. truefoundry/ml/{autogen → _autogen}/client/models/list_artifact_versions_response_dto.py +5 -3
  128. truefoundry/ml/{autogen → _autogen}/client/models/list_artifacts_request_dto.py +3 -3
  129. truefoundry/ml/{autogen → _autogen}/client/models/list_artifacts_response_dto.py +3 -3
  130. truefoundry/ml/{autogen → _autogen}/client/models/list_colums_response_dto.py +3 -3
  131. truefoundry/ml/{autogen → _autogen}/client/models/list_datasets_request_dto.py +2 -2
  132. truefoundry/ml/{autogen → _autogen}/client/models/list_datasets_response_dto.py +3 -3
  133. truefoundry/ml/{autogen → _autogen}/client/models/list_experiments_response_dto.py +3 -3
  134. truefoundry/ml/{autogen → _autogen}/client/models/list_files_for_artifact_version_request_dto.py +2 -2
  135. truefoundry/ml/{autogen → _autogen}/client/models/list_files_for_artifact_versions_response_dto.py +3 -3
  136. truefoundry/ml/{autogen → _autogen}/client/models/list_files_for_dataset_request_dto.py +2 -2
  137. truefoundry/ml/{autogen → _autogen}/client/models/list_files_for_dataset_response_dto.py +3 -3
  138. truefoundry/ml/{autogen → _autogen}/client/models/list_latest_run_logs_response_dto.py +3 -3
  139. truefoundry/ml/{autogen → _autogen}/client/models/list_metric_history_request_dto.py +2 -2
  140. truefoundry/ml/{autogen → _autogen}/client/models/list_metric_history_response_dto.py +3 -3
  141. truefoundry/ml/{autogen → _autogen}/client/models/list_model_version_response_dto.py +3 -3
  142. truefoundry/ml/{autogen → _autogen}/client/models/list_model_versions_request_dto.py +3 -3
  143. truefoundry/ml/{autogen → _autogen}/client/models/list_models_request_dto.py +2 -2
  144. truefoundry/ml/{autogen → _autogen}/client/models/list_models_response_dto.py +3 -3
  145. truefoundry/ml/{autogen → _autogen}/client/models/list_run_artifacts_response_dto.py +3 -3
  146. truefoundry/ml/{autogen → _autogen}/client/models/list_run_logs_response_dto.py +3 -3
  147. truefoundry/ml/_autogen/client/models/local_artifact_source.py +91 -0
  148. truefoundry/ml/_autogen/client/models/local_model_source.py +77 -0
  149. truefoundry/ml/{autogen → _autogen}/client/models/log_batch_request_dto.py +5 -5
  150. truefoundry/ml/{autogen → _autogen}/client/models/log_metric_request_dto.py +2 -2
  151. truefoundry/ml/{autogen → _autogen}/client/models/log_param_request_dto.py +2 -2
  152. truefoundry/ml/_autogen/client/models/manifest.py +229 -0
  153. truefoundry/ml/_autogen/client/models/manifest1.py +217 -0
  154. truefoundry/ml/{autogen/client/models/internal_metadata.py → _autogen/client/models/manifest2.py} +68 -50
  155. truefoundry/ml/{autogen → _autogen}/client/models/method.py +3 -3
  156. truefoundry/ml/{autogen → _autogen}/client/models/metric_collection_dto.py +3 -3
  157. truefoundry/ml/{autogen → _autogen}/client/models/metric_dto.py +2 -2
  158. truefoundry/ml/{autogen → _autogen}/client/models/mime_type.py +3 -3
  159. truefoundry/ml/{autogen → _autogen}/client/models/model_configuration.py +8 -9
  160. truefoundry/ml/{autogen → _autogen}/client/models/model_dto.py +6 -6
  161. truefoundry/ml/{autogen/client/models/model_version_manifest.py → _autogen/client/models/model_manifest.py} +43 -35
  162. truefoundry/ml/{autogen → _autogen}/client/models/model_response_dto.py +3 -3
  163. truefoundry/ml/{autogen → _autogen}/client/models/model_server.py +2 -2
  164. truefoundry/ml/{autogen → _autogen}/client/models/model_version_dto.py +9 -11
  165. truefoundry/ml/{autogen → _autogen}/client/models/model_version_environment.py +5 -7
  166. truefoundry/ml/{autogen → _autogen}/client/models/model_version_response_dto.py +3 -3
  167. truefoundry/ml/{autogen → _autogen}/client/models/multi_part_upload_dto.py +4 -4
  168. truefoundry/ml/{autogen → _autogen}/client/models/multi_part_upload_response_dto.py +3 -3
  169. truefoundry/ml/{autogen → _autogen}/client/models/multi_part_upload_storage_provider.py +2 -2
  170. truefoundry/ml/{autogen → _autogen}/client/models/notify_artifact_version_failure_dto.py +2 -2
  171. truefoundry/ml/{autogen → _autogen}/client/models/onnx_framework.py +4 -6
  172. truefoundry/ml/{autogen → _autogen}/client/models/openapi_spec.py +4 -4
  173. truefoundry/ml/{autogen → _autogen}/client/models/paddle_framework.py +4 -7
  174. truefoundry/ml/{autogen → _autogen}/client/models/param_dto.py +2 -2
  175. truefoundry/ml/{autogen → _autogen}/client/models/parameters.py +4 -4
  176. truefoundry/ml/{autogen → _autogen}/client/models/py_torch_framework.py +4 -7
  177. truefoundry/ml/{autogen → _autogen}/client/models/resolve_agent_app_response_dto.py +3 -3
  178. truefoundry/ml/{autogen → _autogen}/client/models/restore_run_request_dto.py +2 -2
  179. truefoundry/ml/{autogen → _autogen}/client/models/run_data_dto.py +5 -5
  180. truefoundry/ml/{autogen → _autogen}/client/models/run_dto.py +4 -4
  181. truefoundry/ml/{autogen → _autogen}/client/models/run_info_dto.py +3 -3
  182. truefoundry/ml/{autogen → _autogen}/client/models/run_log_dto.py +2 -2
  183. truefoundry/ml/{autogen → _autogen}/client/models/run_log_input_dto.py +2 -2
  184. truefoundry/ml/{autogen → _autogen}/client/models/run_response_dto.py +3 -3
  185. truefoundry/ml/{autogen → _autogen}/client/models/run_tag_dto.py +2 -2
  186. truefoundry/ml/{autogen → _autogen}/client/models/search_runs_request_dto.py +2 -2
  187. truefoundry/ml/{autogen → _autogen}/client/models/search_runs_response_dto.py +3 -3
  188. truefoundry/ml/{autogen → _autogen}/client/models/set_experiment_tag_request_dto.py +2 -2
  189. truefoundry/ml/{autogen → _autogen}/client/models/set_tag_request_dto.py +2 -2
  190. truefoundry/ml/{autogen → _autogen}/client/models/signed_url_dto.py +3 -4
  191. truefoundry/ml/{autogen → _autogen}/client/models/sklearn_framework.py +9 -10
  192. truefoundry/ml/{autogen → _autogen}/client/models/sklearn_model_schema.py +7 -8
  193. truefoundry/ml/{autogen → _autogen}/client/models/sklearn_serialization_format.py +3 -3
  194. truefoundry/ml/{autogen → _autogen}/client/models/source.py +34 -9
  195. truefoundry/ml/{autogen → _autogen}/client/models/source1.py +5 -5
  196. truefoundry/ml/_autogen/client/models/source2.py +177 -0
  197. truefoundry/ml/{autogen → _autogen}/client/models/spa_cy_framework.py +4 -6
  198. truefoundry/ml/{autogen → _autogen}/client/models/stats_models_framework.py +4 -7
  199. truefoundry/ml/{autogen → _autogen}/client/models/stop.py +2 -2
  200. truefoundry/ml/{autogen → _autogen}/client/models/store_run_logs_request_dto.py +3 -3
  201. truefoundry/ml/{autogen → _autogen}/client/models/subject.py +3 -3
  202. truefoundry/ml/{autogen → _autogen}/client/models/subject_type.py +2 -2
  203. truefoundry/ml/{autogen → _autogen}/client/models/system_message.py +6 -8
  204. truefoundry/ml/{autogen → _autogen}/client/models/tensor_flow_framework.py +4 -7
  205. truefoundry/ml/{autogen → _autogen}/client/models/text.py +4 -4
  206. truefoundry/ml/{autogen → _autogen}/client/models/text_content_part.py +5 -5
  207. truefoundry/ml/{autogen → _autogen}/client/models/transformers_framework.py +8 -12
  208. truefoundry/ml/{autogen → _autogen}/client/models/trigger_job_run_config_request_dto.py +3 -3
  209. truefoundry/ml/{autogen → _autogen}/client/models/trigger_job_run_config_response_dto.py +2 -2
  210. truefoundry/ml/{autogen → _autogen}/client/models/true_foundry_managed_source.py +18 -12
  211. truefoundry/ml/{autogen → _autogen}/client/models/update_artifact_version_request_dto.py +5 -7
  212. truefoundry/ml/{autogen → _autogen}/client/models/update_dataset_request_dto.py +13 -3
  213. truefoundry/ml/{autogen → _autogen}/client/models/update_experiment_request_dto.py +2 -2
  214. truefoundry/ml/{autogen → _autogen}/client/models/update_model_version_request_dto.py +5 -7
  215. truefoundry/ml/{autogen → _autogen}/client/models/update_run_request_dto.py +2 -2
  216. truefoundry/ml/{autogen → _autogen}/client/models/update_run_response_dto.py +3 -3
  217. truefoundry/ml/{autogen → _autogen}/client/models/url.py +4 -4
  218. truefoundry/ml/{autogen → _autogen}/client/models/user_message.py +6 -6
  219. truefoundry/ml/{autogen → _autogen}/client/models/validate_external_storage_root_request_dto.py +2 -2
  220. truefoundry/ml/{autogen → _autogen}/client/models/validate_external_storage_root_response_dto.py +2 -2
  221. truefoundry/ml/{autogen → _autogen}/client/models/validation_error.py +3 -3
  222. truefoundry/ml/{autogen → _autogen}/client/models/validation_error_loc_inner.py +2 -2
  223. truefoundry/ml/{autogen → _autogen}/client/models/xg_boost_framework.py +7 -10
  224. truefoundry/ml/{autogen → _autogen}/client/models/xg_boost_model_schema.py +6 -7
  225. truefoundry/ml/{autogen → _autogen}/client/models/xg_boost_serialization_format.py +3 -3
  226. truefoundry/ml/{autogen → _autogen}/client/rest.py +3 -3
  227. truefoundry/ml/_autogen/client_README.md +359 -0
  228. truefoundry/ml/_autogen/entities/artifacts.py +558 -0
  229. truefoundry/ml/artifact/truefoundry_artifact_repo.py +1 -1
  230. truefoundry/ml/cli/commands/model_init.py +1 -1
  231. truefoundry/ml/clients/servicefoundry_client.py +7 -2
  232. truefoundry/ml/entities.py +1 -1
  233. truefoundry/ml/log_types/artifacts/artifact.py +22 -85
  234. truefoundry/ml/log_types/artifacts/dataset.py +5 -5
  235. truefoundry/ml/log_types/artifacts/general_artifact.py +1 -1
  236. truefoundry/ml/log_types/artifacts/model.py +12 -13
  237. truefoundry/ml/log_types/artifacts/utils.py +18 -2
  238. truefoundry/ml/log_types/image/image.py +1 -1
  239. truefoundry/ml/log_types/plot.py +1 -1
  240. truefoundry/ml/mlfoundry_api.py +2 -42
  241. truefoundry/ml/mlfoundry_run.py +1 -1
  242. truefoundry/ml/model_framework.py +3 -3
  243. truefoundry/ml/session.py +7 -4
  244. truefoundry/ml/validation_utils.py +1 -1
  245. truefoundry/workflow/map_task.py +5 -2
  246. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.11rc1.dist-info}/METADATA +2 -1
  247. truefoundry-0.5.11rc1.dist-info/RECORD +377 -0
  248. truefoundry/deploy/auto_gen/models.py +0 -1915
  249. truefoundry/ml/autogen/client/__init__.py +0 -445
  250. truefoundry/ml/autogen/client/api/__init__.py +0 -16
  251. truefoundry/ml/autogen/client/models/__init__.py +0 -416
  252. truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py +0 -69
  253. truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py +0 -69
  254. truefoundry/ml/autogen/client/models/manifest.py +0 -154
  255. truefoundry/ml/autogen/client_README.md +0 -361
  256. truefoundry/ml/autogen/entities/artifacts.py +0 -670
  257. truefoundry-0.5.10rc1.dist-info/RECORD +0 -370
  258. /truefoundry/ml/{autogen → _autogen}/__init__.py +0 -0
  259. /truefoundry/ml/{autogen → _autogen}/client/api_response.py +0 -0
  260. /truefoundry/ml/{autogen → _autogen}/models/__init__.py +0 -0
  261. /truefoundry/ml/{autogen → _autogen}/models/exceptions.py +0 -0
  262. /truefoundry/ml/{autogen → _autogen}/models/schema.py +0 -0
  263. /truefoundry/ml/{autogen → _autogen}/models/signature.py +0 -0
  264. /truefoundry/ml/{autogen → _autogen}/models/utils.py +0 -0
  265. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.11rc1.dist-info}/WHEEL +0 -0
  266. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.11rc1.dist-info}/entry_points.txt +0 -0
@@ -1,416 +0,0 @@
1
- # coding: utf-8
2
-
3
- # flake8: noqa
4
- """
5
- FastAPI
6
-
7
- No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
8
-
9
- The version of the OpenAPI document: 0.1.0
10
- Generated by OpenAPI Generator (https://openapi-generator.tech)
11
-
12
- Do not edit the class manually.
13
- """ # noqa: E501
14
-
15
- # import models into model package
16
- from truefoundry.ml.autogen.client.models.add_custom_metrics_to_model_version_request_dto import (
17
- AddCustomMetricsToModelVersionRequestDto,
18
- )
19
- from truefoundry.ml.autogen.client.models.add_features_to_model_version_request_dto import (
20
- AddFeaturesToModelVersionRequestDto,
21
- )
22
- from truefoundry.ml.autogen.client.models.agent import Agent
23
- from truefoundry.ml.autogen.client.models.agent_app import AgentApp
24
- from truefoundry.ml.autogen.client.models.agent_open_api_tool import AgentOpenAPITool
25
- from truefoundry.ml.autogen.client.models.agent_open_api_tool_with_fqn import (
26
- AgentOpenAPIToolWithFQN,
27
- )
28
- from truefoundry.ml.autogen.client.models.agent_with_fqn import AgentWithFQN
29
- from truefoundry.ml.autogen.client.models.artifact_dto import ArtifactDto
30
- from truefoundry.ml.autogen.client.models.artifact_response_dto import (
31
- ArtifactResponseDto,
32
- )
33
- from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
34
- from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
35
- from truefoundry.ml.autogen.client.models.artifact_version_manifest import (
36
- ArtifactVersionManifest,
37
- )
38
- from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
39
- ArtifactVersionResponseDto,
40
- )
41
- from truefoundry.ml.autogen.client.models.artifact_version_serialization_format import (
42
- ArtifactVersionSerializationFormat,
43
- )
44
- from truefoundry.ml.autogen.client.models.artifact_version_status import (
45
- ArtifactVersionStatus,
46
- )
47
- from truefoundry.ml.autogen.client.models.assistant_message import AssistantMessage
48
- from truefoundry.ml.autogen.client.models.authorize_user_for_model_request_dto import (
49
- AuthorizeUserForModelRequestDto,
50
- )
51
- from truefoundry.ml.autogen.client.models.authorize_user_for_model_version_request_dto import (
52
- AuthorizeUserForModelVersionRequestDto,
53
- )
54
- from truefoundry.ml.autogen.client.models.blob_storage_reference import (
55
- BlobStorageReference,
56
- )
57
- from truefoundry.ml.autogen.client.models.body_get_search_runs_get import (
58
- BodyGetSearchRunsGet,
59
- )
60
- from truefoundry.ml.autogen.client.models.chat_prompt import ChatPrompt
61
- from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
62
- ChatPromptMessagesInner,
63
- )
64
- from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
65
- from truefoundry.ml.autogen.client.models.command import Command
66
- from truefoundry.ml.autogen.client.models.content import Content
67
- from truefoundry.ml.autogen.client.models.content1 import Content1
68
- from truefoundry.ml.autogen.client.models.content2 import Content2
69
- from truefoundry.ml.autogen.client.models.content2_any_of_inner import (
70
- Content2AnyOfInner,
71
- )
72
- from truefoundry.ml.autogen.client.models.create_artifact_request_dto import (
73
- CreateArtifactRequestDto,
74
- )
75
- from truefoundry.ml.autogen.client.models.create_artifact_response_dto import (
76
- CreateArtifactResponseDto,
77
- )
78
- from truefoundry.ml.autogen.client.models.create_artifact_version_request_dto import (
79
- CreateArtifactVersionRequestDto,
80
- )
81
- from truefoundry.ml.autogen.client.models.create_artifact_version_response_dto import (
82
- CreateArtifactVersionResponseDto,
83
- )
84
- from truefoundry.ml.autogen.client.models.create_dataset_request_dto import (
85
- CreateDatasetRequestDto,
86
- )
87
- from truefoundry.ml.autogen.client.models.create_experiment_request_dto import (
88
- CreateExperimentRequestDto,
89
- )
90
- from truefoundry.ml.autogen.client.models.create_experiment_response_dto import (
91
- CreateExperimentResponseDto,
92
- )
93
- from truefoundry.ml.autogen.client.models.create_model_version_request_dto import (
94
- CreateModelVersionRequestDto,
95
- )
96
- from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_request_dto import (
97
- CreateMultiPartUploadForDatasetRequestDto,
98
- )
99
- from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_response_dto import (
100
- CreateMultiPartUploadForDatasetResponseDto,
101
- )
102
- from truefoundry.ml.autogen.client.models.create_multi_part_upload_request_dto import (
103
- CreateMultiPartUploadRequestDto,
104
- )
105
- from truefoundry.ml.autogen.client.models.create_python_deployment_config_request_dto import (
106
- CreatePythonDeploymentConfigRequestDto,
107
- )
108
- from truefoundry.ml.autogen.client.models.create_python_deployment_config_response_dto import (
109
- CreatePythonDeploymentConfigResponseDto,
110
- )
111
- from truefoundry.ml.autogen.client.models.create_run_request_dto import (
112
- CreateRunRequestDto,
113
- )
114
- from truefoundry.ml.autogen.client.models.create_run_response_dto import (
115
- CreateRunResponseDto,
116
- )
117
- from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
118
- CreateWorkflowTaskConfigRequestDto,
119
- )
120
- from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
121
- from truefoundry.ml.autogen.client.models.dataset_response_dto import DatasetResponseDto
122
- from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto import (
123
- DeleteArtifactVersionsRequestDto,
124
- )
125
- from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
126
- DeleteDatasetRequestDto,
127
- )
128
- from truefoundry.ml.autogen.client.models.delete_files_for_dataset_request_dto import (
129
- DeleteFilesForDatasetRequestDto,
130
- )
131
- from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
132
- DeleteModelVersionRequestDto,
133
- )
134
- from truefoundry.ml.autogen.client.models.delete_run_request import DeleteRunRequest
135
- from truefoundry.ml.autogen.client.models.delete_tag_request_dto import (
136
- DeleteTagRequestDto,
137
- )
138
- from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
139
- from truefoundry.ml.autogen.client.models.experiment_id_request_dto import (
140
- ExperimentIdRequestDto,
141
- )
142
- from truefoundry.ml.autogen.client.models.experiment_response_dto import (
143
- ExperimentResponseDto,
144
- )
145
- from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
146
- from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto import (
147
- ExportDeploymentFilesRequestDto,
148
- )
149
- from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
150
- ExternalBlobStorageSource,
151
- )
152
- from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
153
- from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
154
- from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
155
- FinalizeArtifactVersionRequestDto,
156
- )
157
- from truefoundry.ml.autogen.client.models.framework import Framework
158
- from truefoundry.ml.autogen.client.models.get_artifact_version_aliases_response_dto import (
159
- GetArtifactVersionAliasesResponseDto,
160
- )
161
- from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
162
- GetExperimentResponseDto,
163
- )
164
- from truefoundry.ml.autogen.client.models.get_latest_run_log_response_dto import (
165
- GetLatestRunLogResponseDto,
166
- )
167
- from truefoundry.ml.autogen.client.models.get_metric_history_response import (
168
- GetMetricHistoryResponse,
169
- )
170
- from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
171
- GetSignedURLForDatasetWriteRequestDto,
172
- )
173
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_request_dto import (
174
- GetSignedURLsForArtifactVersionReadRequestDto,
175
- )
176
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_response_dto import (
177
- GetSignedURLsForArtifactVersionReadResponseDto,
178
- )
179
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_request_dto import (
180
- GetSignedURLsForArtifactVersionWriteRequestDto,
181
- )
182
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_response_dto import (
183
- GetSignedURLsForArtifactVersionWriteResponseDto,
184
- )
185
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_request_dto import (
186
- GetSignedURLsForDatasetReadRequestDto,
187
- )
188
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_response_dto import (
189
- GetSignedURLsForDatasetReadResponseDto,
190
- )
191
- from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_response_dto import (
192
- GetSignedURLsForDatasetWriteResponseDto,
193
- )
194
- from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
195
- GetTenantIdResponseDto,
196
- )
197
- from truefoundry.ml.autogen.client.models.gluon_framework import GluonFramework
198
- from truefoundry.ml.autogen.client.models.h2_o_framework import H2OFramework
199
- from truefoundry.ml.autogen.client.models.http_validation_error import (
200
- HTTPValidationError,
201
- )
202
- from truefoundry.ml.autogen.client.models.image_content_part import ImageContentPart
203
- from truefoundry.ml.autogen.client.models.image_url import ImageUrl
204
- from truefoundry.ml.autogen.client.models.infer_method_name import InferMethodName
205
- from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
206
- from truefoundry.ml.autogen.client.models.keras_framework import KerasFramework
207
- from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
208
- from truefoundry.ml.autogen.client.models.library_name import LibraryName
209
- from truefoundry.ml.autogen.client.models.light_gbm_framework import LightGBMFramework
210
- from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
211
- ListArtifactVersionsRequestDto,
212
- )
213
- from truefoundry.ml.autogen.client.models.list_artifact_versions_response_dto import (
214
- ListArtifactVersionsResponseDto,
215
- )
216
- from truefoundry.ml.autogen.client.models.list_artifacts_request_dto import (
217
- ListArtifactsRequestDto,
218
- )
219
- from truefoundry.ml.autogen.client.models.list_artifacts_response_dto import (
220
- ListArtifactsResponseDto,
221
- )
222
- from truefoundry.ml.autogen.client.models.list_colums_response_dto import (
223
- ListColumsResponseDto,
224
- )
225
- from truefoundry.ml.autogen.client.models.list_datasets_request_dto import (
226
- ListDatasetsRequestDto,
227
- )
228
- from truefoundry.ml.autogen.client.models.list_datasets_response_dto import (
229
- ListDatasetsResponseDto,
230
- )
231
- from truefoundry.ml.autogen.client.models.list_experiments_response_dto import (
232
- ListExperimentsResponseDto,
233
- )
234
- from truefoundry.ml.autogen.client.models.list_files_for_artifact_version_request_dto import (
235
- ListFilesForArtifactVersionRequestDto,
236
- )
237
- from truefoundry.ml.autogen.client.models.list_files_for_artifact_versions_response_dto import (
238
- ListFilesForArtifactVersionsResponseDto,
239
- )
240
- from truefoundry.ml.autogen.client.models.list_files_for_dataset_request_dto import (
241
- ListFilesForDatasetRequestDto,
242
- )
243
- from truefoundry.ml.autogen.client.models.list_files_for_dataset_response_dto import (
244
- ListFilesForDatasetResponseDto,
245
- )
246
- from truefoundry.ml.autogen.client.models.list_latest_run_logs_response_dto import (
247
- ListLatestRunLogsResponseDto,
248
- )
249
- from truefoundry.ml.autogen.client.models.list_metric_history_request_dto import (
250
- ListMetricHistoryRequestDto,
251
- )
252
- from truefoundry.ml.autogen.client.models.list_metric_history_response_dto import (
253
- ListMetricHistoryResponseDto,
254
- )
255
- from truefoundry.ml.autogen.client.models.list_model_version_response_dto import (
256
- ListModelVersionResponseDto,
257
- )
258
- from truefoundry.ml.autogen.client.models.list_model_versions_request_dto import (
259
- ListModelVersionsRequestDto,
260
- )
261
- from truefoundry.ml.autogen.client.models.list_models_request_dto import (
262
- ListModelsRequestDto,
263
- )
264
- from truefoundry.ml.autogen.client.models.list_models_response_dto import (
265
- ListModelsResponseDto,
266
- )
267
- from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import (
268
- ListRunArtifactsResponseDto,
269
- )
270
- from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
271
- ListRunLogsResponseDto,
272
- )
273
- from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
274
- LogBatchRequestDto,
275
- )
276
- from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
277
- LogMetricRequestDto,
278
- )
279
- from truefoundry.ml.autogen.client.models.log_param_request_dto import (
280
- LogParamRequestDto,
281
- )
282
- from truefoundry.ml.autogen.client.models.manifest import Manifest
283
- from truefoundry.ml.autogen.client.models.method import Method
284
- from truefoundry.ml.autogen.client.models.metric_collection_dto import (
285
- MetricCollectionDto,
286
- )
287
- from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
288
- from truefoundry.ml.autogen.client.models.mime_type import MimeType
289
- from truefoundry.ml.autogen.client.models.model_configuration import ModelConfiguration
290
- from truefoundry.ml.autogen.client.models.model_dto import ModelDto
291
- from truefoundry.ml.autogen.client.models.model_response_dto import ModelResponseDto
292
- from truefoundry.ml.autogen.client.models.model_server import ModelServer
293
- from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
294
- from truefoundry.ml.autogen.client.models.model_version_environment import (
295
- ModelVersionEnvironment,
296
- )
297
- from truefoundry.ml.autogen.client.models.model_version_manifest import (
298
- ModelVersionManifest,
299
- )
300
- from truefoundry.ml.autogen.client.models.model_version_response_dto import (
301
- ModelVersionResponseDto,
302
- )
303
- from truefoundry.ml.autogen.client.models.multi_part_upload_dto import (
304
- MultiPartUploadDto,
305
- )
306
- from truefoundry.ml.autogen.client.models.multi_part_upload_response_dto import (
307
- MultiPartUploadResponseDto,
308
- )
309
- from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider import (
310
- MultiPartUploadStorageProvider,
311
- )
312
- from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
313
- NotifyArtifactVersionFailureDto,
314
- )
315
- from truefoundry.ml.autogen.client.models.onnx_framework import ONNXFramework
316
- from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
317
- from truefoundry.ml.autogen.client.models.paddle_framework import PaddleFramework
318
- from truefoundry.ml.autogen.client.models.param_dto import ParamDto
319
- from truefoundry.ml.autogen.client.models.parameters import Parameters
320
- from truefoundry.ml.autogen.client.models.py_torch_framework import PyTorchFramework
321
- from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
322
- ResolveAgentAppResponseDto,
323
- )
324
- from truefoundry.ml.autogen.client.models.restore_run_request_dto import (
325
- RestoreRunRequestDto,
326
- )
327
- from truefoundry.ml.autogen.client.models.run_data_dto import RunDataDto
328
- from truefoundry.ml.autogen.client.models.run_dto import RunDto
329
- from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
330
- from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
331
- from truefoundry.ml.autogen.client.models.run_log_input_dto import RunLogInputDto
332
- from truefoundry.ml.autogen.client.models.run_response_dto import RunResponseDto
333
- from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
334
- from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
335
- SearchRunsRequestDto,
336
- )
337
- from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
338
- SearchRunsResponseDto,
339
- )
340
- from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import (
341
- SetExperimentTagRequestDto,
342
- )
343
- from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
344
- from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
345
- from truefoundry.ml.autogen.client.models.sklearn_framework import SklearnFramework
346
- from truefoundry.ml.autogen.client.models.sklearn_model_schema import SklearnModelSchema
347
- from truefoundry.ml.autogen.client.models.sklearn_serialization_format import (
348
- SklearnSerializationFormat,
349
- )
350
- from truefoundry.ml.autogen.client.models.source import Source
351
- from truefoundry.ml.autogen.client.models.source1 import Source1
352
- from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
353
- from truefoundry.ml.autogen.client.models.stats_models_framework import (
354
- StatsModelsFramework,
355
- )
356
- from truefoundry.ml.autogen.client.models.stop import Stop
357
- from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
358
- StoreRunLogsRequestDto,
359
- )
360
- from truefoundry.ml.autogen.client.models.subject import Subject
361
- from truefoundry.ml.autogen.client.models.subject_type import SubjectType
362
- from truefoundry.ml.autogen.client.models.system_message import SystemMessage
363
- from truefoundry.ml.autogen.client.models.tensor_flow_framework import (
364
- TensorFlowFramework,
365
- )
366
- from truefoundry.ml.autogen.client.models.text import Text
367
- from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
368
- from truefoundry.ml.autogen.client.models.transformers_framework import (
369
- TransformersFramework,
370
- )
371
- from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto import (
372
- TriggerJobRunConfigRequestDto,
373
- )
374
- from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
375
- TriggerJobRunConfigResponseDto,
376
- )
377
- from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
378
- TrueFoundryManagedSource,
379
- )
380
- from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
381
- UpdateArtifactVersionRequestDto,
382
- )
383
- from truefoundry.ml.autogen.client.models.update_dataset_request_dto import (
384
- UpdateDatasetRequestDto,
385
- )
386
- from truefoundry.ml.autogen.client.models.update_experiment_request_dto import (
387
- UpdateExperimentRequestDto,
388
- )
389
- from truefoundry.ml.autogen.client.models.update_model_version_request_dto import (
390
- UpdateModelVersionRequestDto,
391
- )
392
- from truefoundry.ml.autogen.client.models.update_run_request_dto import (
393
- UpdateRunRequestDto,
394
- )
395
- from truefoundry.ml.autogen.client.models.update_run_response_dto import (
396
- UpdateRunResponseDto,
397
- )
398
- from truefoundry.ml.autogen.client.models.url import Url
399
- from truefoundry.ml.autogen.client.models.user_message import UserMessage
400
- from truefoundry.ml.autogen.client.models.validate_external_storage_root_request_dto import (
401
- ValidateExternalStorageRootRequestDto,
402
- )
403
- from truefoundry.ml.autogen.client.models.validate_external_storage_root_response_dto import (
404
- ValidateExternalStorageRootResponseDto,
405
- )
406
- from truefoundry.ml.autogen.client.models.validation_error import ValidationError
407
- from truefoundry.ml.autogen.client.models.validation_error_loc_inner import (
408
- ValidationErrorLocInner,
409
- )
410
- from truefoundry.ml.autogen.client.models.xg_boost_framework import XGBoostFramework
411
- from truefoundry.ml.autogen.client.models.xg_boost_model_schema import (
412
- XGBoostModelSchema,
413
- )
414
- from truefoundry.ml.autogen.client.models.xg_boost_serialization_format import (
415
- XGBoostSerializationFormat,
416
- )
@@ -1,69 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- FastAPI
5
-
6
- No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
-
8
- The version of the OpenAPI document: 0.1.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
- from __future__ import annotations
15
-
16
- import json
17
- import pprint
18
- import re # noqa: F401
19
- from typing import Any, Dict, Optional
20
-
21
- from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
22
-
23
-
24
- class AddCustomMetricsToModelVersionRequestDto(BaseModel):
25
- """
26
- AddCustomMetricsToModelVersionRequestDto
27
- """
28
-
29
- id: StrictStr = Field(...)
30
- custom_metrics: Optional[conlist(Dict[str, Any])] = None
31
- __properties = ["id", "custom_metrics"]
32
-
33
- class Config:
34
- """Pydantic configuration"""
35
-
36
- allow_population_by_field_name = True
37
- validate_assignment = True
38
-
39
- def to_str(self) -> str:
40
- """Returns the string representation of the model using alias"""
41
- return pprint.pformat(self.dict(by_alias=True))
42
-
43
- def to_json(self) -> str:
44
- """Returns the JSON representation of the model using alias"""
45
- return json.dumps(self.to_dict())
46
-
47
- @classmethod
48
- def from_json(cls, json_str: str) -> AddCustomMetricsToModelVersionRequestDto:
49
- """Create an instance of AddCustomMetricsToModelVersionRequestDto from a JSON string"""
50
- return cls.from_dict(json.loads(json_str))
51
-
52
- def to_dict(self):
53
- """Returns the dictionary representation of the model using alias"""
54
- _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
55
- return _dict
56
-
57
- @classmethod
58
- def from_dict(cls, obj: dict) -> AddCustomMetricsToModelVersionRequestDto:
59
- """Create an instance of AddCustomMetricsToModelVersionRequestDto from a dict"""
60
- if obj is None:
61
- return None
62
-
63
- if not isinstance(obj, dict):
64
- return AddCustomMetricsToModelVersionRequestDto.parse_obj(obj)
65
-
66
- _obj = AddCustomMetricsToModelVersionRequestDto.parse_obj(
67
- {"id": obj.get("id"), "custom_metrics": obj.get("custom_metrics")}
68
- )
69
- return _obj
@@ -1,69 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- FastAPI
5
-
6
- No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
-
8
- The version of the OpenAPI document: 0.1.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
- from __future__ import annotations
15
-
16
- import json
17
- import pprint
18
- import re # noqa: F401
19
- from typing import Any, Dict, Optional
20
-
21
- from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
22
-
23
-
24
- class AddFeaturesToModelVersionRequestDto(BaseModel):
25
- """
26
- AddFeaturesToModelVersionRequestDto
27
- """
28
-
29
- id: StrictStr = Field(...)
30
- features: Optional[conlist(Dict[str, Any])] = None
31
- __properties = ["id", "features"]
32
-
33
- class Config:
34
- """Pydantic configuration"""
35
-
36
- allow_population_by_field_name = True
37
- validate_assignment = True
38
-
39
- def to_str(self) -> str:
40
- """Returns the string representation of the model using alias"""
41
- return pprint.pformat(self.dict(by_alias=True))
42
-
43
- def to_json(self) -> str:
44
- """Returns the JSON representation of the model using alias"""
45
- return json.dumps(self.to_dict())
46
-
47
- @classmethod
48
- def from_json(cls, json_str: str) -> AddFeaturesToModelVersionRequestDto:
49
- """Create an instance of AddFeaturesToModelVersionRequestDto from a JSON string"""
50
- return cls.from_dict(json.loads(json_str))
51
-
52
- def to_dict(self):
53
- """Returns the dictionary representation of the model using alias"""
54
- _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
55
- return _dict
56
-
57
- @classmethod
58
- def from_dict(cls, obj: dict) -> AddFeaturesToModelVersionRequestDto:
59
- """Create an instance of AddFeaturesToModelVersionRequestDto from a dict"""
60
- if obj is None:
61
- return None
62
-
63
- if not isinstance(obj, dict):
64
- return AddFeaturesToModelVersionRequestDto.parse_obj(obj)
65
-
66
- _obj = AddFeaturesToModelVersionRequestDto.parse_obj(
67
- {"id": obj.get("id"), "features": obj.get("features")}
68
- )
69
- return _obj