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