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
@@ -17,20 +17,18 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
20
+ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
21
21
 
22
22
 
23
23
  class ExternalBlobStorageSource(BaseModel):
24
24
  """
25
- +label=External Blob Storage Source # noqa: E501
25
+ ExternalBlobStorageSource
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=..., description="+label=Type +usage=Type of the source +value=external"
30
- )
31
- uri: StrictStr = Field(
28
+ type: StrictStr = Field(default=..., description="Type of the source")
29
+ uri: constr(strict=True) = Field(
32
30
  default=...,
33
- description="+label=URI +usage=URI referencing a path in the blob storage bucket linked to the MLRepo",
31
+ description="URI referencing a path in the blob storage bucket linked to the MLRepo",
34
32
  )
35
33
  __properties = ["type", "uri"]
36
34
 
@@ -41,6 +39,15 @@ class ExternalBlobStorageSource(BaseModel):
41
39
  raise ValueError("must be one of enum values ('external')")
42
40
  return value
43
41
 
42
+ @validator("uri")
43
+ def uri_validate_regular_expression(cls, value):
44
+ """Validates the regular expression"""
45
+ if not re.match(r"^(s3|gs|wasbs):\/\/.*", value):
46
+ raise ValueError(
47
+ r"must validate the regular expression /^(s3|gs|wasbs):\/\/.*/"
48
+ )
49
+ return value
50
+
44
51
  class Config:
45
52
  """Pydantic configuration"""
46
53
 
@@ -22,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class FastAIFramework(BaseModel):
24
24
  """
25
- +label=FastAI +icon=fastai # noqa: E501
25
+ FastAIFramework
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=...,
30
- description="+label=Type +usage=Type of the framework +value=fastai",
31
- )
28
+ type: StrictStr = Field(default=..., description="Type of the framework")
32
29
  __properties = ["type"]
33
30
 
34
31
  @validator("type")
@@ -16,6 +16,7 @@ from __future__ import annotations
16
16
  import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
+ from datetime import datetime
19
20
  from typing import Optional
20
21
 
21
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictBool, StrictInt, StrictStr
@@ -30,7 +31,8 @@ class FileInfoDto(BaseModel):
30
31
  is_dir: StrictBool = Field(...)
31
32
  file_size: Optional[StrictInt] = None
32
33
  signed_url: Optional[StrictStr] = None
33
- __properties = ["path", "is_dir", "file_size", "signed_url"]
34
+ last_modified: Optional[datetime] = None
35
+ __properties = ["path", "is_dir", "file_size", "signed_url", "last_modified"]
34
36
 
35
37
  class Config:
36
38
  """Pydantic configuration"""
@@ -71,6 +73,7 @@ class FileInfoDto(BaseModel):
71
73
  "is_dir": obj.get("is_dir"),
72
74
  "file_size": obj.get("file_size"),
73
75
  "signed_url": obj.get("signed_url"),
76
+ "last_modified": obj.get("last_modified"),
74
77
  }
75
78
  )
76
79
  return _obj
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Any, Dict, Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.manifest1 import Manifest1
21
+ from truefoundry.ml._autogen.client.models.manifest1 import Manifest1
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
23
23
 
24
24
 
@@ -18,26 +18,26 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import TYPE_CHECKING, Any, List, Optional, Union
20
20
 
21
- from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
22
- from truefoundry.ml.autogen.client.models.gluon_framework import GluonFramework
23
- from truefoundry.ml.autogen.client.models.h2_o_framework import H2OFramework
24
- from truefoundry.ml.autogen.client.models.keras_framework import KerasFramework
25
- from truefoundry.ml.autogen.client.models.light_gbm_framework import LightGBMFramework
26
- from truefoundry.ml.autogen.client.models.onnx_framework import ONNXFramework
27
- from truefoundry.ml.autogen.client.models.paddle_framework import PaddleFramework
28
- from truefoundry.ml.autogen.client.models.py_torch_framework import PyTorchFramework
29
- from truefoundry.ml.autogen.client.models.sklearn_framework import SklearnFramework
30
- from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
31
- from truefoundry.ml.autogen.client.models.stats_models_framework import (
21
+ from truefoundry.ml._autogen.client.models.fast_ai_framework import FastAIFramework
22
+ from truefoundry.ml._autogen.client.models.gluon_framework import GluonFramework
23
+ from truefoundry.ml._autogen.client.models.h2_o_framework import H2OFramework
24
+ from truefoundry.ml._autogen.client.models.keras_framework import KerasFramework
25
+ from truefoundry.ml._autogen.client.models.light_gbm_framework import LightGBMFramework
26
+ from truefoundry.ml._autogen.client.models.onnx_framework import ONNXFramework
27
+ from truefoundry.ml._autogen.client.models.paddle_framework import PaddleFramework
28
+ from truefoundry.ml._autogen.client.models.py_torch_framework import PyTorchFramework
29
+ from truefoundry.ml._autogen.client.models.sklearn_framework import SklearnFramework
30
+ from truefoundry.ml._autogen.client.models.spa_cy_framework import SpaCyFramework
31
+ from truefoundry.ml._autogen.client.models.stats_models_framework import (
32
32
  StatsModelsFramework,
33
33
  )
34
- from truefoundry.ml.autogen.client.models.tensor_flow_framework import (
34
+ from truefoundry.ml._autogen.client.models.tensor_flow_framework import (
35
35
  TensorFlowFramework,
36
36
  )
37
- from truefoundry.ml.autogen.client.models.transformers_framework import (
37
+ from truefoundry.ml._autogen.client.models.transformers_framework import (
38
38
  TransformersFramework,
39
39
  )
40
- from truefoundry.ml.autogen.client.models.xg_boost_framework import XGBoostFramework
40
+ from truefoundry.ml._autogen.client.models.xg_boost_framework import XGBoostFramework
41
41
  from truefoundry.pydantic_v1 import (
42
42
  BaseModel,
43
43
  Field,
@@ -65,7 +65,7 @@ FRAMEWORK_ANY_OF_SCHEMAS = [
65
65
 
66
66
  class Framework(BaseModel):
67
67
  """
68
- +label=Framework +usage=Framework for the model version like Transformers, PyTorch, Sklearn, Xgboost etc with framework specific metadata. This will be used to infer model deployment configuration +uiProps={\"descriptionInline\":true}
68
+ Framework for the model version like Transformers, PyTorch, Sklearn, Xgboost etc with framework specific metadata. This will be used to infer model deployment configuration
69
69
  """
70
70
 
71
71
  # data type: TransformersFramework
@@ -18,8 +18,8 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
22
- from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
21
+ from truefoundry.ml._autogen.client.models.experiment_dto import ExperimentDto
22
+ from truefoundry.ml._autogen.client.models.run_info_dto import RunInfoDto
23
23
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
24
24
 
25
25
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
20
+ from truefoundry.ml._autogen.client.models.run_log_dto import RunLogDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field
22
22
 
23
23
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
20
+ from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
22
22
 
23
23
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
20
+ from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
22
22
 
23
23
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
20
+ from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
22
22
 
23
23
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
20
+ from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
22
22
 
23
23
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
20
+ from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
22
22
 
23
23
 
@@ -22,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class GluonFramework(BaseModel):
24
24
  """
25
- +label=Gluon +icon=gluon # noqa: E501
25
+ GluonFramework
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=..., description="+label=Type +usage=Type of the framework +value=gluon"
30
- )
28
+ type: StrictStr = Field(default=..., description="Type of the framework")
31
29
  __properties = ["type"]
32
30
 
33
31
  @validator("type")
@@ -22,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class H2OFramework(BaseModel):
24
24
  """
25
- +label=H2O +icon=h2o # noqa: E501
25
+ H2OFramework
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=..., description="+label=Type +usage=Type of the framework +value=h2o"
30
- )
28
+ type: StrictStr = Field(default=..., description="Type of the framework")
31
29
  __properties = ["type"]
32
30
 
33
31
  @validator("type")
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.validation_error import ValidationError
21
+ from truefoundry.ml._autogen.client.models.validation_error import ValidationError
22
22
  from truefoundry.pydantic_v1 import BaseModel, conlist
23
23
 
24
24
 
@@ -17,21 +17,18 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.image_url import ImageUrl
20
+ from truefoundry.ml._autogen.client.models.image_url import ImageUrl
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
 
24
24
  class ImageContentPart(BaseModel):
25
25
  """
26
- +label=Image Content +usage=Image content for the message # noqa: E501
26
+ Image content for the message # noqa: E501
27
27
  """
28
28
 
29
- type: StrictStr = Field(
30
- default=..., description="+value=image_url +usage=Type of the content part"
31
- )
29
+ type: StrictStr = Field(default=..., description="Type of the content part")
32
30
  image_url: ImageUrl = Field(
33
- default=...,
34
- description="+label=Image URL +usage=Image URL linking to the image",
31
+ default=..., description="Image URL linking to the image"
35
32
  )
36
33
  __properties = ["type", "image_url"]
37
34
 
@@ -17,13 +17,13 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.url import Url
20
+ from truefoundry.ml._autogen.client.models.url import Url
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field
22
22
 
23
23
 
24
24
  class ImageUrl(BaseModel):
25
25
  """
26
- +label=Image URL +usage=Image URL linking to the image # noqa: E501
26
+ Image URL linking to the image # noqa: E501
27
27
  """
28
28
 
29
29
  url: Url = Field(...)
@@ -19,7 +19,7 @@ from aenum import Enum
19
19
 
20
20
  class InferMethodName(str, Enum):
21
21
  """
22
- +label=Inference Method Name +usage=Name of the method used for inference
22
+ Name of the method used for inference
23
23
  """
24
24
 
25
25
  """
@@ -22,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class KerasFramework(BaseModel):
24
24
  """
25
- +label=Keras +icon=keras # noqa: E501
25
+ KerasFramework
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=..., description="+label=Type +usage=Type of the framework +value=keras"
30
- )
28
+ type: StrictStr = Field(default=..., description="Type of the framework")
31
29
  __properties = ["type"]
32
30
 
33
31
  @validator("type")
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
21
+ from truefoundry.ml._autogen.client.models.run_log_dto import RunLogDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, conlist
23
23
 
24
24
 
@@ -19,7 +19,7 @@ from aenum import Enum
19
19
 
20
20
  class LibraryName(str, Enum):
21
21
  """
22
- +label=Library Name +usage=Name of the library for the framework
22
+ Name of the library for the framework
23
23
  """
24
24
 
25
25
  """
@@ -22,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class LightGBMFramework(BaseModel):
24
24
  """
25
- +label=LightGBM +icon=lightbgm # noqa: E501
25
+ LightGBMFramework
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=...,
30
- description="+label=Type +usage=Type of the framework +value=lightgbm",
31
- )
28
+ type: StrictStr = Field(default=..., description="Type of the framework")
32
29
  __properties = ["type"]
33
30
 
34
31
  @validator("type")
@@ -18,8 +18,8 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
22
- from truefoundry.ml.autogen.client.models.artifact_version_status import (
21
+ from truefoundry.ml._autogen.client.models.artifact_type import ArtifactType
22
+ from truefoundry.ml._autogen.client.models.artifact_version_status import (
23
23
  ArtifactVersionStatus,
24
24
  )
25
25
  from truefoundry.pydantic_v1 import BaseModel, StrictBool, StrictInt, StrictStr, conlist
@@ -18,7 +18,9 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
21
+ from truefoundry.ml._autogen.client.models.artifact_version_dto import (
22
+ ArtifactVersionDto,
23
+ )
22
24
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
23
25
 
24
26
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
21
+ from truefoundry.ml._autogen.client.models.artifact_type import ArtifactType
22
22
  from truefoundry.pydantic_v1 import BaseModel, StrictBool, StrictInt, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.artifact_dto import ArtifactDto
21
+ from truefoundry.ml._autogen.client.models.artifact_dto import ArtifactDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
23
23
 
24
24
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
20
+ from truefoundry.ml._autogen.client.models.columns_dto import ColumnsDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field
22
22
 
23
23
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
21
+ from truefoundry.ml._autogen.client.models.dataset_dto import DatasetDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
21
+ from truefoundry.ml._autogen.client.models.experiment_dto import ExperimentDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
21
+ from truefoundry.ml._autogen.client.models.file_info_dto import FileInfoDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
21
+ from truefoundry.ml._autogen.client.models.file_info_dto import FileInfoDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
23
23
 
24
24
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
20
+ from truefoundry.ml._autogen.client.models.latest_run_log_dto import LatestRunLogDto
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
22
22
 
23
23
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.metric_collection_dto import (
20
+ from truefoundry.ml._autogen.client.models.metric_collection_dto import (
21
21
  MetricCollectionDto,
22
22
  )
23
23
  from truefoundry.pydantic_v1 import BaseModel, Field, conlist
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
21
+ from truefoundry.ml._autogen.client.models.model_version_dto import ModelVersionDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.artifact_version_status import (
21
+ from truefoundry.ml._autogen.client.models.artifact_version_status import (
22
22
  ArtifactVersionStatus,
23
23
  )
24
24
  from truefoundry.pydantic_v1 import BaseModel, StrictBool, StrictInt, StrictStr, conlist
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.model_dto import ModelDto
21
+ from truefoundry.ml._autogen.client.models.model_dto import ModelDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
21
+ from truefoundry.ml._autogen.client.models.file_info_dto import FileInfoDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
23
23
 
24
24
 
@@ -18,7 +18,7 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
21
+ from truefoundry.ml._autogen.client.models.run_log_dto import RunLogDto
22
22
  from truefoundry.pydantic_v1 import BaseModel, conlist
23
23
 
24
24
 
@@ -17,7 +17,7 @@ import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
19
 
20
- from truefoundry.ml.autogen.client.models.artifact_path import ArtifactPath
20
+ from truefoundry.ml._autogen.client.models.artifact_path import ArtifactPath
21
21
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist, validator
22
22
 
23
23
 
@@ -26,12 +26,10 @@ class LocalArtifactSource(BaseModel):
26
26
  LocalArtifactSource
27
27
  """
28
28
 
29
- type: StrictStr = Field(
30
- default=..., description="+label=Type +usage=Type of the source +value=local"
31
- )
29
+ type: StrictStr = Field(default=..., description="Type of the source")
32
30
  paths: conlist(ArtifactPath) = Field(
33
31
  default=...,
34
- description="+label=Paths +usage=Array of ArtifactPath objects representing the source and destination paths",
32
+ description="Array of ArtifactPath objects representing the source and destination paths",
35
33
  )
36
34
  __properties = ["type", "paths"]
37
35
 
@@ -25,12 +25,9 @@ class LocalModelSource(BaseModel):
25
25
  LocalModelSource
26
26
  """
27
27
 
28
- type: StrictStr = Field(
29
- default=..., description="+label=Type +usage=Type of the source +value=local"
30
- )
28
+ type: StrictStr = Field(default=..., description="Type of the source")
31
29
  file_or_folder: StrictStr = Field(
32
- default=...,
33
- description="+label=Path to the model file or folder +usage=Path to the model file or folder",
30
+ default=..., description="Path to the model file or folder"
34
31
  )
35
32
  __properties = ["type", "file_or_folder"]
36
33
 
@@ -18,9 +18,9 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
22
- from truefoundry.ml.autogen.client.models.param_dto import ParamDto
23
- from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
21
+ from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
22
+ from truefoundry.ml._autogen.client.models.param_dto import ParamDto
23
+ from truefoundry.ml._autogen.client.models.run_tag_dto import RunTagDto
24
24
  from truefoundry.pydantic_v1 import BaseModel, StrictStr, conlist
25
25
 
26
26