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