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