truefoundry 0.5.7rc1__tar.gz → 0.5.8rc1__tar.gz

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 (373) hide show
  1. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/PKG-INFO +1 -1
  2. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/pyproject.toml +24 -12
  3. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/__main__.py +2 -2
  4. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/credential_provider.py +1 -1
  5. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/entities.py +2 -4
  6. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/servicefoundry_client.py +7 -2
  7. truefoundry-0.5.8rc1/truefoundry/common/types.py +7 -0
  8. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/utils.py +1 -1
  9. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/deploy_command.py +3 -5
  10. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/clients/servicefoundry_client.py +3 -6
  11. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/deploy.py +2 -2
  12. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/deploy_workflow.py +8 -5
  13. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/deployable_patched_models.py +13 -9
  14. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/models.py +1 -1
  15. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/patched_models.py +1 -1
  16. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/source.py +2 -1
  17. truefoundry-0.5.7rc1/truefoundry/workflow/example/deploy.sh +0 -1
  18. truefoundry-0.5.7rc1/truefoundry/workflow/example/hello_world_package/workflow.py +0 -20
  19. truefoundry-0.5.7rc1/truefoundry/workflow/example/package/test_workflow.py +0 -151
  20. truefoundry-0.5.7rc1/truefoundry/workflow/example/truefoundry.yaml +0 -9
  21. truefoundry-0.5.7rc1/truefoundry/workflow/example/workflow.yaml +0 -116
  22. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/README.md +0 -0
  23. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/__init__.py +0 -0
  24. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/__init__.py +0 -0
  25. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/agents/__init__.py +0 -0
  26. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/agents/base.py +0 -0
  27. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/agents/developer.py +0 -0
  28. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/agents/project_identifier.py +0 -0
  29. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/agents/tester.py +0 -0
  30. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/cli.py +0 -0
  31. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/constants.py +0 -0
  32. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/exception.py +0 -0
  33. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/logger.py +0 -0
  34. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/__init__.py +0 -0
  35. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/ask.py +0 -0
  36. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/base.py +0 -0
  37. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/commit.py +0 -0
  38. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/docker_build.py +0 -0
  39. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/docker_run.py +0 -0
  40. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/file_type_counts.py +0 -0
  41. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/list_files.py +0 -0
  42. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/read_file.py +0 -0
  43. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/send_request.py +0 -0
  44. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/tools/write_file.py +0 -0
  45. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/utils/client.py +0 -0
  46. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/utils/diff.py +0 -0
  47. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/autodeploy/utils/pydantic_compat.py +0 -0
  48. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/__init__.py +0 -0
  49. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/config.py +0 -0
  50. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/console.py +0 -0
  51. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/const.py +0 -0
  52. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/display_util.py +0 -0
  53. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/cli/util.py +0 -0
  54. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/__init__.py +0 -0
  55. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/auth_service_client.py +0 -0
  56. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/constants.py +0 -0
  57. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/credential_file_manager.py +0 -0
  58. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/exceptions.py +0 -0
  59. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/request_utils.py +0 -0
  60. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/session.py +0 -0
  61. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/storage_provider_utils.py +0 -0
  62. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/common/warnings.py +0 -0
  63. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/__init__.py +0 -0
  64. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/auto_gen/models.py +0 -0
  65. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/__init__.py +0 -0
  66. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/builders/__init__.py +0 -0
  67. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/builders/dockerfile.py +0 -0
  68. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py +0 -0
  69. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py +0 -0
  70. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py +0 -0
  71. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py +0 -0
  72. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/constants.py +0 -0
  73. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/docker_service.py +0 -0
  74. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/builder/utils.py +0 -0
  75. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/__init__.py +0 -0
  76. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/__init__.py +0 -0
  77. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/apply_command.py +0 -0
  78. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/build_command.py +0 -0
  79. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/create_command.py +0 -0
  80. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/delete_command.py +0 -0
  81. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/deploy_init_command.py +0 -0
  82. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/get_command.py +0 -0
  83. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/list_command.py +0 -0
  84. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/login_command.py +0 -0
  85. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/logout_command.py +0 -0
  86. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/logs_command.py +0 -0
  87. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/patch_application_command.py +0 -0
  88. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/patch_command.py +0 -0
  89. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/redeploy_command.py +0 -0
  90. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/terminate_comand.py +0 -0
  91. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/cli/commands/trigger_command.py +0 -0
  92. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/core/__init__.py +0 -0
  93. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/core/login.py +0 -0
  94. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/core/logout.py +0 -0
  95. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/io/__init__.py +0 -0
  96. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/io/output_callback.py +0 -0
  97. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/io/rich_output_callback.py +0 -0
  98. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/__init__.py +0 -0
  99. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/clients/__init__.py +0 -0
  100. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/const.py +0 -0
  101. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/dao/__init__.py +0 -0
  102. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/dao/application.py +0 -0
  103. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/dao/apply.py +0 -0
  104. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/dao/version.py +0 -0
  105. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/dao/workspace.py +0 -0
  106. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/logs_utils.py +0 -0
  107. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/messages.py +0 -0
  108. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/model/__init__.py +0 -0
  109. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/model/entity.py +0 -0
  110. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/session.py +0 -0
  111. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/util.py +0 -0
  112. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/lib/win32.py +0 -0
  113. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/python_deploy_codegen.py +0 -0
  114. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/__init__.py +0 -0
  115. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/deploy/v2/lib/__init__.py +0 -0
  116. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/logger.py +0 -0
  117. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/__init__.py +0 -0
  118. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/artifact/__init__.py +0 -0
  119. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/artifact/truefoundry_artifact_repo.py +0 -0
  120. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/__init__.py +0 -0
  121. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/__init__.py +0 -0
  122. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/__init__.py +0 -0
  123. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/auth_api.py +0 -0
  124. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/deprecated_api.py +0 -0
  125. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/experiments_api.py +0 -0
  126. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/generate_code_snippet_api.py +0 -0
  127. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/health_api.py +0 -0
  128. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/metrics_api.py +0 -0
  129. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py +0 -0
  130. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/run_artifacts_api.py +0 -0
  131. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api/runs_api.py +0 -0
  132. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api_client.py +0 -0
  133. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/api_response.py +0 -0
  134. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/configuration.py +0 -0
  135. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/exceptions.py +0 -0
  136. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/__init__.py +0 -0
  137. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py +0 -0
  138. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py +0 -0
  139. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/agent.py +0 -0
  140. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/agent_app.py +0 -0
  141. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/agent_open_api_tool.py +0 -0
  142. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py +0 -0
  143. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/agent_with_fqn.py +0 -0
  144. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_dto.py +0 -0
  145. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_response_dto.py +0 -0
  146. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_type.py +0 -0
  147. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_version_dto.py +0 -0
  148. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_version_manifest.py +0 -0
  149. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_version_response_dto.py +0 -0
  150. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py +0 -0
  151. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/artifact_version_status.py +0 -0
  152. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/assistant_message.py +0 -0
  153. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py +0 -0
  154. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py +0 -0
  155. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/blob_storage_reference.py +0 -0
  156. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/body_get_search_runs_get.py +0 -0
  157. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/chat_prompt.py +0 -0
  158. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/chat_prompt_messages_inner.py +0 -0
  159. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/columns_dto.py +0 -0
  160. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/command.py +0 -0
  161. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/content.py +0 -0
  162. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/content1.py +0 -0
  163. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/content2.py +0 -0
  164. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/content2_any_of_inner.py +0 -0
  165. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_artifact_request_dto.py +0 -0
  166. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_artifact_response_dto.py +0 -0
  167. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py +0 -0
  168. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py +0 -0
  169. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_dataset_request_dto.py +0 -0
  170. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_experiment_request_dto.py +0 -0
  171. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_experiment_response_dto.py +0 -0
  172. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_model_version_request_dto.py +0 -0
  173. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_request_dto.py +0 -0
  174. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_response_dto.py +0 -0
  175. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py +0 -0
  176. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py +0 -0
  177. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py +0 -0
  178. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_run_request_dto.py +0 -0
  179. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_run_response_dto.py +0 -0
  180. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py +0 -0
  181. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/dataset_dto.py +0 -0
  182. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/dataset_response_dto.py +0 -0
  183. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py +0 -0
  184. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py +0 -0
  185. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/delete_files_for_dataset_request_dto.py +0 -0
  186. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py +0 -0
  187. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/delete_run_request.py +0 -0
  188. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/delete_tag_request_dto.py +0 -0
  189. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/experiment_dto.py +0 -0
  190. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/experiment_id_request_dto.py +0 -0
  191. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/experiment_response_dto.py +0 -0
  192. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/experiment_tag_dto.py +0 -0
  193. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/export_deployment_files_request_dto.py +0 -0
  194. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/external_blob_storage_source.py +0 -0
  195. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/fast_ai_framework.py +0 -0
  196. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/file_info_dto.py +0 -0
  197. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py +0 -0
  198. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/framework.py +0 -0
  199. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_artifact_version_aliases_response_dto.py +0 -0
  200. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_experiment_response_dto.py +0 -0
  201. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py +0 -0
  202. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_metric_history_response.py +0 -0
  203. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_request_dto.py +0 -0
  204. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_request_dto.py +0 -0
  205. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_response_dto.py +0 -0
  206. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_request_dto.py +0 -0
  207. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_response_dto.py +0 -0
  208. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dto.py +0 -0
  209. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py +0 -0
  210. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py +0 -0
  211. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py +0 -0
  212. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/gluon_framework.py +0 -0
  213. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/h2_o_framework.py +0 -0
  214. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/http_validation_error.py +0 -0
  215. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/image_content_part.py +0 -0
  216. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/image_url.py +0 -0
  217. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/infer_method_name.py +0 -0
  218. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/internal_metadata.py +0 -0
  219. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/keras_framework.py +0 -0
  220. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/latest_run_log_dto.py +0 -0
  221. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/library_name.py +0 -0
  222. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/light_gbm_framework.py +0 -0
  223. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py +0 -0
  224. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py +0 -0
  225. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py +0 -0
  226. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_artifacts_response_dto.py +0 -0
  227. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_colums_response_dto.py +0 -0
  228. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_datasets_request_dto.py +0 -0
  229. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_datasets_response_dto.py +0 -0
  230. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_experiments_response_dto.py +0 -0
  231. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_files_for_artifact_version_request_dto.py +0 -0
  232. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_files_for_artifact_versions_response_dto.py +0 -0
  233. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_files_for_dataset_request_dto.py +0 -0
  234. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_files_for_dataset_response_dto.py +0 -0
  235. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_latest_run_logs_response_dto.py +0 -0
  236. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_metric_history_request_dto.py +0 -0
  237. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_metric_history_response_dto.py +0 -0
  238. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_model_version_response_dto.py +0 -0
  239. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_model_versions_request_dto.py +0 -0
  240. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_models_request_dto.py +0 -0
  241. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_models_response_dto.py +0 -0
  242. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_run_artifacts_response_dto.py +0 -0
  243. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py +0 -0
  244. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/log_batch_request_dto.py +0 -0
  245. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/log_metric_request_dto.py +0 -0
  246. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/log_param_request_dto.py +0 -0
  247. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/manifest.py +0 -0
  248. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/method.py +0 -0
  249. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/metric_collection_dto.py +0 -0
  250. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/metric_dto.py +0 -0
  251. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/mime_type.py +0 -0
  252. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_configuration.py +0 -0
  253. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_dto.py +0 -0
  254. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_response_dto.py +0 -0
  255. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_server.py +0 -0
  256. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_version_dto.py +0 -0
  257. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_version_environment.py +0 -0
  258. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_version_manifest.py +0 -0
  259. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/model_version_response_dto.py +0 -0
  260. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/multi_part_upload_dto.py +0 -0
  261. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py +0 -0
  262. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/multi_part_upload_storage_provider.py +0 -0
  263. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/notify_artifact_version_failure_dto.py +0 -0
  264. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/onnx_framework.py +0 -0
  265. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/openapi_spec.py +0 -0
  266. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/paddle_framework.py +0 -0
  267. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/param_dto.py +0 -0
  268. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/parameters.py +0 -0
  269. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/py_torch_framework.py +0 -0
  270. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py +0 -0
  271. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/restore_run_request_dto.py +0 -0
  272. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_data_dto.py +0 -0
  273. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_dto.py +0 -0
  274. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_info_dto.py +0 -0
  275. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_log_dto.py +0 -0
  276. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_log_input_dto.py +0 -0
  277. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_response_dto.py +0 -0
  278. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/run_tag_dto.py +0 -0
  279. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/search_runs_request_dto.py +0 -0
  280. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/search_runs_response_dto.py +0 -0
  281. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py +0 -0
  282. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/set_tag_request_dto.py +0 -0
  283. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/signed_url_dto.py +0 -0
  284. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/sklearn_framework.py +0 -0
  285. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/sklearn_model_schema.py +0 -0
  286. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/sklearn_serialization_format.py +0 -0
  287. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/source.py +0 -0
  288. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/source1.py +0 -0
  289. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/spa_cy_framework.py +0 -0
  290. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/stats_models_framework.py +0 -0
  291. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/stop.py +0 -0
  292. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py +0 -0
  293. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/subject.py +0 -0
  294. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/subject_type.py +0 -0
  295. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/system_message.py +0 -0
  296. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/tensor_flow_framework.py +0 -0
  297. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/text.py +0 -0
  298. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/text_content_part.py +0 -0
  299. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/transformers_framework.py +0 -0
  300. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py +0 -0
  301. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py +0 -0
  302. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/true_foundry_managed_source.py +0 -0
  303. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py +0 -0
  304. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/update_dataset_request_dto.py +0 -0
  305. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/update_experiment_request_dto.py +0 -0
  306. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/update_model_version_request_dto.py +0 -0
  307. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/update_run_request_dto.py +0 -0
  308. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/update_run_response_dto.py +0 -0
  309. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/url.py +0 -0
  310. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/user_message.py +0 -0
  311. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/validate_external_storage_root_request_dto.py +0 -0
  312. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/validate_external_storage_root_response_dto.py +0 -0
  313. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/validation_error.py +0 -0
  314. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/validation_error_loc_inner.py +0 -0
  315. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/xg_boost_framework.py +0 -0
  316. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/xg_boost_model_schema.py +0 -0
  317. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py +0 -0
  318. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client/rest.py +0 -0
  319. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/client_README.md +0 -0
  320. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/entities/artifacts.py +0 -0
  321. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/models/__init__.py +0 -0
  322. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/models/exceptions.py +0 -0
  323. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/models/schema.py +0 -0
  324. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/models/signature.py +0 -0
  325. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/autogen/models/utils.py +0 -0
  326. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/cli/__init__.py +0 -0
  327. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/cli/cli.py +0 -0
  328. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/cli/commands/__init__.py +0 -0
  329. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/cli/commands/download.py +0 -0
  330. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/cli/commands/model_init.py +0 -0
  331. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/cli/utils.py +0 -0
  332. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/clients/__init__.py +0 -0
  333. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/clients/servicefoundry_client.py +0 -0
  334. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/constants.py +0 -0
  335. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/entities.py +0 -0
  336. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/enums.py +0 -0
  337. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/exceptions.py +0 -0
  338. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/git_info.py +0 -0
  339. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/internal_namespace.py +0 -0
  340. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/__init__.py +0 -0
  341. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/artifacts/artifact.py +0 -0
  342. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/artifacts/constants.py +0 -0
  343. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/artifacts/dataset.py +0 -0
  344. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/artifacts/general_artifact.py +0 -0
  345. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/artifacts/model.py +0 -0
  346. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/artifacts/utils.py +0 -0
  347. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/image/__init__.py +0 -0
  348. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/image/constants.py +0 -0
  349. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/image/image.py +0 -0
  350. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/image/image_normalizer.py +0 -0
  351. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/image/types.py +0 -0
  352. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/plot.py +0 -0
  353. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/pydantic_base.py +0 -0
  354. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/log_types/utils.py +0 -0
  355. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/logger.py +0 -0
  356. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/mlfoundry_api.py +0 -0
  357. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/mlfoundry_run.py +0 -0
  358. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/model_framework.py +0 -0
  359. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/run_utils.py +0 -0
  360. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/session.py +0 -0
  361. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/ml/validation_utils.py +0 -0
  362. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/pydantic_v1.py +0 -0
  363. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/version.py +0 -0
  364. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/__init__.py +0 -0
  365. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/container_task.py +0 -0
  366. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/map_task.py +0 -0
  367. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/python_task.py +0 -0
  368. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/remote_filesystem/__init__.py +0 -0
  369. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/remote_filesystem/logger.py +0 -0
  370. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py +0 -0
  371. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py +0 -0
  372. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/task.py +0 -0
  373. {truefoundry-0.5.7rc1 → truefoundry-0.5.8rc1}/truefoundry/workflow/workflow.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: truefoundry
3
- Version: 0.5.7rc1
3
+ Version: 0.5.8rc1
4
4
  Summary: TrueFoundry CLI
5
5
  Author: Abhishek Choudhary
6
6
  Author-email: abhishek@truefoundry.com
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "truefoundry"
3
- version = "0.5.7rc1"
3
+ version = "0.5.8rc1"
4
4
  description = "TrueFoundry CLI"
5
5
  authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
6
6
  readme = "README.md"
@@ -55,10 +55,11 @@ workflow = ["flytekit"]
55
55
  [tool.poetry.group.dev.dependencies]
56
56
  Pillow = ">=9.1.1,<11.0.0"
57
57
  ipykernel = ">=6.16.0"
58
+ ipywidgets = ">=8.0.0"
58
59
  ipython = ">=7.10.0"
59
60
  langchain = ">=0.2.0,<0.3.0"
60
61
  matplotlib = ">=3.0.0,<4.0.0"
61
- mypy = "*"
62
+ mypy = ">=1.14.1"
62
63
  plotly = ">=5.8.2,<6.0.0"
63
64
  pandas = [
64
65
  # This is split into two because poetry will select a version which has no wheel on python >= 3.10 :/
@@ -68,7 +69,7 @@ pandas = [
68
69
  pre-commit = ">=3.5.0,<4.0.0"
69
70
  pytest = ">=8.2.2,<9.0.0"
70
71
  pytest-env = ">=1.1.5,<2.0.0"
71
- ruff = ">=0.4.10"
72
+ ruff = ">=0.9.4"
72
73
  seaborn = ">=0.13.0,<0.14.0"
73
74
  scikit-learn = ">=1.3.0,<2.0.0"
74
75
  shap = [
@@ -98,9 +99,19 @@ lint.select = [
98
99
  "B", # flake8-bugbear
99
100
  "Q", # flake8-quotes
100
101
  "I",
102
+ "T10",
103
+ "ASYNC",
104
+ ]
105
+ lint.ignore = [
106
+ "E501",
107
+ ]
108
+ exclude = [
109
+ ".venv",
110
+ "venv",
111
+ "repos",
112
+ "truefoundry/ml/autogen/**",
113
+ "truefoundry/deploy/auto_gen/**",
101
114
  ]
102
- lint.ignore = ["E501"]
103
- exclude = ["venv", "repos"]
104
115
 
105
116
  [tool.ruff.lint.per-file-ignores]
106
117
  "__init__.py" = ["F401"]
@@ -110,7 +121,7 @@ exclude = ["venv", "repos"]
110
121
 
111
122
  [tool.mypy]
112
123
  plugins = [
113
- "pydantic.mypy"
124
+ "pydantic.v1.mypy"
114
125
  ]
115
126
  install_types = true
116
127
  non_interactive = true
@@ -119,11 +130,11 @@ allow_redefinition = false
119
130
  allow_untyped_globals = false
120
131
  check_untyped_defs = true
121
132
  disallow_any_generics = true
122
- disallow_incomplete_defs = false
133
+ disallow_incomplete_defs = true
123
134
  disallow_subclassing_any = true
124
- disallow_untyped_calls = false
125
- disallow_untyped_decorators = false
126
- disallow_untyped_defs = false
135
+ disallow_untyped_calls = true
136
+ disallow_untyped_decorators = true
137
+ disallow_untyped_defs = true
127
138
  follow_imports = "silent"
128
139
  force_uppercase_builtins = true
129
140
  no_implicit_optional = true
@@ -137,16 +148,17 @@ warn_unreachable = true
137
148
  warn_unused_ignores = true
138
149
 
139
150
  packages = [
140
- "truefoundry.common",
141
- "truefoundry.ml",
151
+ "truefoundry",
142
152
  ]
143
153
  exclude = [
144
154
  "^truefoundry/ml/autogen",
145
155
  "^truefoundry/deploy/auto_gen",
146
156
  "^truefoundry/pydantic_v1.py$",
157
+ "^truefoundry/autodeploy/",
147
158
  ]
148
159
  disable_error_code = [
149
160
  "import-untyped",
161
+ "method-assign",
150
162
  ]
151
163
 
152
164
  [tool.pydantic-mypy]
@@ -71,7 +71,7 @@ def truefoundry_cli(ctx, json, debug):
71
71
  logger.add_cli_handler(level=log_level)
72
72
 
73
73
 
74
- def create_truefoundry_cli() -> click.MultiCommand:
74
+ def create_truefoundry_cli() -> click.Group:
75
75
  """Generates CLI by combining all subcommands into a main CLI and returns in
76
76
 
77
77
  Returns:
@@ -105,7 +105,7 @@ def main():
105
105
  try:
106
106
  cli = create_truefoundry_cli()
107
107
  except Exception as e:
108
- raise click.exceptions.UsageError(message=str(e)) from e
108
+ raise click.UsageError(message=str(e)) from e
109
109
  sys.exit(cli())
110
110
 
111
111
 
@@ -38,7 +38,7 @@ class EnvCredentialProvider(CredentialProvider):
38
38
  )
39
39
  self._host = resolve_tfy_host()
40
40
  self._auth_service = AuthServiceClient.from_tfy_host(tfy_host=self._host)
41
- self._token: Token = Token(access_token=api_key, refresh_token=None) # type: ignore[call-arg]
41
+ self._token: Token = Token(access_token=api_key, refresh_token=None)
42
42
 
43
43
  @staticmethod
44
44
  def can_provide() -> bool:
@@ -62,7 +62,7 @@ class Token(BaseModel):
62
62
  return (exp - time.time()) < buffer_in_seconds
63
63
 
64
64
  def to_user_info(self) -> UserInfo:
65
- return UserInfo( # type: ignore[call-arg]
65
+ return UserInfo(
66
66
  user_id=self.decoded_value["username"],
67
67
  email=self.decoded_value["email"]
68
68
  if "email" in self.decoded_value
@@ -81,9 +81,7 @@ class CredentialsFileContent(BaseModel):
81
81
  host: NonEmptyStr
82
82
 
83
83
  def to_token(self) -> Token:
84
- return Token( # type: ignore[call-arg]
85
- access_token=self.access_token, refresh_token=self.refresh_token
86
- )
84
+ return Token(access_token=self.access_token, refresh_token=self.refresh_token)
87
85
 
88
86
 
89
87
  class TenantInfo(BaseModel):
@@ -1,9 +1,11 @@
1
1
  from __future__ import annotations
2
2
 
3
3
  import functools
4
+ from typing import Callable, TypeVar
4
5
 
5
6
  import requests
6
7
  from packaging import version
8
+ from typing_extensions import ParamSpec
7
9
 
8
10
  from truefoundry.common.constants import (
9
11
  SERVICEFOUNDRY_CLIENT_MAX_RETRIES,
@@ -24,10 +26,13 @@ from truefoundry.common.utils import (
24
26
  from truefoundry.logger import logger
25
27
  from truefoundry.version import __version__
26
28
 
29
+ P = ParamSpec("P")
30
+ R = TypeVar("R")
27
31
 
28
- def check_min_cli_version(fn):
32
+
33
+ def check_min_cli_version(fn: Callable[P, R]) -> Callable[P, R]:
29
34
  @functools.wraps(fn)
30
- def inner(*args, **kwargs):
35
+ def inner(*args: P.args, **kwargs: P.kwargs) -> R:
31
36
  client: "ServiceFoundryServiceClient" = args[0]
32
37
  client.check_min_cli_version()
33
38
  return fn(*args, **kwargs)
@@ -0,0 +1,7 @@
1
+ from typing import Protocol
2
+
3
+
4
+ class UploadCodePackageCallable(Protocol):
5
+ def __call__(
6
+ self, workspace_fqn: str, component_name: str, package_local_path: str
7
+ ) -> str: ...
@@ -53,7 +53,7 @@ def get_tfy_servers_config(tfy_host: str) -> _TFYServersConfig:
53
53
  global _tfy_servers_config
54
54
  if _tfy_servers_config is None:
55
55
  if ENV_VARS.TFY_CLI_LOCAL_DEV_MODE:
56
- _tfy_servers_config = _TFYServersConfig() # type: ignore[call-arg]
56
+ _tfy_servers_config = _TFYServersConfig()
57
57
  else:
58
58
  _tfy_servers_config = _TFYServersConfig.from_tfy_host(tfy_host)
59
59
  return _tfy_servers_config
@@ -55,7 +55,7 @@ def _get_default_spec_file():
55
55
  help="Wait and tail the deployment progress",
56
56
  )
57
57
  @click.option(
58
- "--force-deploy/--no-force-deploy",
58
+ "--force/--no-force",
59
59
  is_flag=True,
60
60
  show_default=True,
61
61
  default=False,
@@ -68,7 +68,7 @@ def deploy_command(
68
68
  file: str,
69
69
  workspace_fqn: Optional[str],
70
70
  wait: bool,
71
- force_deploy: bool,
71
+ force: bool = False,
72
72
  ):
73
73
  if ctx.invoked_subcommand is not None:
74
74
  return
@@ -85,9 +85,7 @@ def deploy_command(
85
85
  application_definition = yaml.safe_load(f)
86
86
 
87
87
  application = Application.parse_obj(application_definition)
88
- application.deploy(
89
- workspace_fqn=workspace_fqn, wait=wait, force_deploy=force_deploy
90
- )
88
+ application.deploy(workspace_fqn=workspace_fqn, wait=wait, force=force)
91
89
  sys.exit(0)
92
90
 
93
91
  click.echo(
@@ -4,7 +4,7 @@ import json
4
4
  import os
5
5
  import time
6
6
  from datetime import datetime, timezone
7
- from typing import TYPE_CHECKING, Any, Dict, List, Optional
7
+ from typing import Any, Dict, List, Optional
8
8
  from urllib.parse import urljoin
9
9
 
10
10
  import requests
@@ -50,9 +50,6 @@ DEPLOYMENT_LOGS_SUBSCRIBE_MESSAGE = "DEPLOYMENT_LOGS"
50
50
  BUILD_LOGS_SUBSCRIBE_MESSAGE = "BUILD_LOGS"
51
51
  MAX_RETRIES_WORKFLOW_TRIGGER = 3
52
52
 
53
- if TYPE_CHECKING:
54
- from truefoundry.deploy.auto_gen.models import Application
55
-
56
53
 
57
54
  def _upload_packaged_code(metadata, package_file):
58
55
  file_size = os.stat(package_file).st_size
@@ -242,13 +239,13 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
242
239
  self,
243
240
  workspace_id: str,
244
241
  application: auto_gen_models.Workflow,
245
- force_deploy: bool,
242
+ force: bool = False,
246
243
  ) -> Deployment:
247
244
  data = {
248
245
  "workspaceId": workspace_id,
249
246
  "name": application.name,
250
247
  "manifest": application.dict(exclude_none=True),
251
- "forceDeploy": force_deploy,
248
+ "forceDeploy": force,
252
249
  }
253
250
  logger.debug(json.dumps(data))
254
251
  url = f"{self._api_server_url}/{VERSION_PREFIX}/deployment"
@@ -252,7 +252,7 @@ def deploy_component(
252
252
  component: Component,
253
253
  workspace_fqn: Optional[str] = None,
254
254
  wait: bool = True,
255
- force_deploy: bool = False,
255
+ force: bool = False,
256
256
  ) -> Deployment:
257
257
  _warn_when_gpu_selected_without_cuda(component=component)
258
258
  workspace_fqn = _resolve_workspace_fqn(
@@ -274,7 +274,7 @@ def deploy_component(
274
274
  response = client.deploy_application(
275
275
  workspace_id=workspace_id,
276
276
  application=updated_component,
277
- force_deploy=force_deploy,
277
+ force=force,
278
278
  )
279
279
  logger.info(
280
280
  "🚀 Deployment started for application '%s'. Deployment FQN is '%s'.",
@@ -1,7 +1,7 @@
1
1
  import os
2
2
  import sys
3
3
  from pathlib import Path
4
- from typing import Callable, Dict, List, Optional, Union
4
+ from typing import Dict, List, Optional, Union
5
5
 
6
6
  import requirements
7
7
  from flytekit.configuration import (
@@ -19,6 +19,7 @@ from flytekit.tools.translator import TaskSpec as FlyteTaskSpec
19
19
  from flytekit.tools.translator import WorkflowSpec as FlyteWorkflowSpec
20
20
  from google.protobuf.json_format import MessageToDict
21
21
 
22
+ from truefoundry.common.types import UploadCodePackageCallable
22
23
  from truefoundry.deploy.auto_gen import models as auto_gen_models
23
24
  from truefoundry.deploy.lib.clients.servicefoundry_client import (
24
25
  ServiceFoundryServiceClient,
@@ -39,7 +40,7 @@ from truefoundry.workflow.workflow import (
39
40
  def _handle_code_upload_for_workflow(
40
41
  workflow: auto_gen_models.Workflow,
41
42
  workspace_fqn: str,
42
- upload_code_package: Callable[[str, str, str], str],
43
+ upload_code_package: UploadCodePackageCallable,
43
44
  ) -> auto_gen_models.Workflow:
44
45
  new_workflow = workflow.copy(deep=True)
45
46
  new_workflow.source = local_source_to_remote_source(
@@ -92,7 +93,7 @@ def _is_tfy_wf_present_in_task_python_build(
92
93
 
93
94
 
94
95
  def _is_dynamic_task(flyte_task: FlyteTaskSpec) -> bool:
95
- envs: Dict[str:str] = flyte_task.template.container.env or {}
96
+ envs: Dict[str, str] = flyte_task.template.container.env or {}
96
97
  return SERIALIZED_CONTEXT_ENV_VAR in envs.keys()
97
98
 
98
99
 
@@ -273,7 +274,7 @@ def deploy_workflow(
273
274
  workflow: auto_gen_models.Workflow,
274
275
  workspace_fqn: str,
275
276
  wait: bool = True,
276
- force_deploy: bool = False,
277
+ force: bool = False,
277
278
  ) -> Deployment:
278
279
  _generate_manifest_for_workflow(workflow)
279
280
  _validate_workspace_fqn(workflow, workspace_fqn)
@@ -296,7 +297,9 @@ def deploy_workflow(
296
297
  )
297
298
 
298
299
  deployment = client.deploy_application(
299
- workspace_id=workspace_id, application=workflow
300
+ workspace_id=workspace_id,
301
+ application=workflow,
302
+ force=force,
300
303
  )
301
304
  logger.info(
302
305
  "🚀 Deployment started for application '%s'. Deployment FQN is '%s'.",
@@ -12,13 +12,13 @@ class DeployablePatchedModelBase(BaseModel):
12
12
  extra = "forbid"
13
13
 
14
14
  def deploy(
15
- self, workspace_fqn: str, wait: bool = True, force_deploy: bool = False
15
+ self, workspace_fqn: str, wait: bool = True, force: bool = False
16
16
  ) -> Deployment:
17
17
  return deploy_component(
18
18
  component=self,
19
19
  workspace_fqn=workspace_fqn,
20
20
  wait=wait,
21
- force_deploy=force_deploy,
21
+ force=force,
22
22
  )
23
23
 
24
24
 
@@ -26,7 +26,7 @@ class Service(models.Service, DeployablePatchedModelBase):
26
26
  type: Literal["service"] = "service"
27
27
  resources: models.Resources = Field(default_factory=models.Resources)
28
28
  # This is being patched because cue export marks this as a "number"
29
- replicas: Union[conint(ge=0, le=100), models.ServiceAutoscaling] = Field(
29
+ replicas: Union[conint(ge=0, le=100), models.ServiceAutoscaling] = Field( # type: ignore[valid-type]
30
30
  1,
31
31
  description="+label=Replicas\n+usage=Replicas of service you want to run\n+icon=fa-clone\n+sort=3",
32
32
  )
@@ -70,7 +70,7 @@ class ApplicationSet(models.ApplicationSet, DeployablePatchedModelBase):
70
70
 
71
71
  class AsyncService(models.AsyncService, DeployablePatchedModelBase):
72
72
  type: Literal["async-service"] = "async-service"
73
- replicas: Union[conint(ge=0, le=100), models.AsyncServiceAutoscaling] = 1
73
+ replicas: Union[conint(ge=0, le=100), models.AsyncServiceAutoscaling] = 1 # type: ignore[valid-type]
74
74
  resources: models.Resources = Field(default_factory=models.Resources)
75
75
 
76
76
 
@@ -81,7 +81,7 @@ class SSHServer(models.SSHServer, DeployablePatchedModelBase):
81
81
 
82
82
  class Application(models.Application, DeployablePatchedModelBase):
83
83
  def deploy(
84
- self, workspace_fqn: str, wait: bool = True, force_deploy: bool = False
84
+ self, workspace_fqn: str, wait: bool = True, force: bool = False
85
85
  ) -> Deployment:
86
86
  if isinstance(self.__root__, models.Workflow):
87
87
  from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
@@ -90,14 +90,14 @@ class Application(models.Application, DeployablePatchedModelBase):
90
90
  workflow=self.__root__,
91
91
  workspace_fqn=workspace_fqn,
92
92
  wait=wait,
93
- force_deploy=force_deploy,
93
+ force=force,
94
94
  )
95
95
  else:
96
96
  return deploy_component(
97
97
  component=self.__root__,
98
98
  workspace_fqn=workspace_fqn,
99
99
  wait=wait,
100
- force_deploy=force_deploy,
100
+ force=force,
101
101
  )
102
102
 
103
103
 
@@ -107,7 +107,11 @@ class Workflow(models.Workflow, DeployablePatchedModelBase):
107
107
  default_factory=lambda: LocalSource(local_build=False)
108
108
  )
109
109
 
110
- def deploy(self, workspace_fqn: str, wait: bool = True) -> Deployment:
110
+ def deploy(
111
+ self, workspace_fqn: str, wait: bool = True, force: bool = False
112
+ ) -> Deployment:
111
113
  from truefoundry.deploy.v2.lib.deploy_workflow import deploy_workflow
112
114
 
113
- return deploy_workflow(workflow=self, workspace_fqn=workspace_fqn, wait=wait)
115
+ return deploy_workflow(
116
+ workflow=self, workspace_fqn=workspace_fqn, wait=wait, force=force
117
+ )
@@ -26,7 +26,7 @@ class BuildResponse(BaseModel):
26
26
 
27
27
 
28
28
  class AppDeploymentStatusResponse(BaseModel):
29
- state: create_model(
29
+ state: create_model( # type: ignore[valid-type]
30
30
  "State",
31
31
  isTerminalState=(bool, ...),
32
32
  type=(str, ...),
@@ -518,7 +518,7 @@ class ArtifactsDownload(models.ArtifactsDownload, PatchedModelBase):
518
518
 
519
519
  class NvidiaGPU(models.NvidiaGPU, PatchedModelBase):
520
520
  type: Literal["nvidia_gpu"] = "nvidia_gpu"
521
- name: Optional[Union[GPUType, constr(regex=r"^tpu-[a-z\d\-]+$")]] = None
521
+ name: Optional[Union[GPUType, constr(regex=r"^tpu-[a-z\d\-]+$")]] = None # type: ignore[valid-type]
522
522
 
523
523
 
524
524
  class NvidiaMIGGPU(models.NvidiaMIGGPU, PatchedModelBase):
@@ -8,6 +8,7 @@ from typing import Callable, List, Optional
8
8
  import gitignorefile
9
9
  from tqdm import tqdm
10
10
 
11
+ from truefoundry.common.types import UploadCodePackageCallable
11
12
  from truefoundry.common.warnings import TrueFoundryDeprecationWarning
12
13
  from truefoundry.deploy import builder
13
14
  from truefoundry.deploy.auto_gen import models
@@ -136,7 +137,7 @@ def local_source_to_remote_source(
136
137
  local_source: models.LocalSource,
137
138
  workspace_fqn: str,
138
139
  component_name: str,
139
- upload_code_package: Callable[[str, str, str], str],
140
+ upload_code_package: UploadCodePackageCallable,
140
141
  ) -> RemoteSource:
141
142
  with tempfile.TemporaryDirectory() as local_dir:
142
143
  package_local_path = os.path.join(local_dir, "build.tar.gz")
@@ -1 +0,0 @@
1
- tfy deploy --workspace_fqn tfy-ctl-euwe1-devtest:nikhil-ws
@@ -1,20 +0,0 @@
1
- from truefoundry.workflow import PythonTaskConfig, TaskPythonBuild, task, workflow
2
-
3
-
4
- @task(
5
- task_config=PythonTaskConfig(
6
- image=TaskPythonBuild(
7
- python_version="3.9",
8
- pip_packages=["truefoundry[workflow]"],
9
- ),
10
- service_account="tfy-workflows-sa",
11
- )
12
- )
13
- def say_hello() -> str:
14
- return "Hello, World!"
15
-
16
-
17
- @workflow
18
- def hello_world_wf() -> str:
19
- res = say_hello()
20
- return res
@@ -1,151 +0,0 @@
1
- import os
2
- from functools import partial
3
- from pathlib import Path
4
- from typing import List, Optional, Tuple
5
-
6
- from truefoundry.deploy import Image, NvidiaGPU, Resources
7
- from truefoundry.workflow import (
8
- ContainerTask,
9
- ContainerTaskConfig,
10
- FlyteDirectory,
11
- PythonTaskConfig,
12
- TaskPythonBuild,
13
- conditional,
14
- map_task,
15
- task,
16
- workflow,
17
- )
18
-
19
- cpu_task_config = PythonTaskConfig(
20
- image=TaskPythonBuild(
21
- python_version="3.9",
22
- pip_packages=["truefoundry[workflow]==0.4.1"],
23
- ),
24
- resources=Resources(cpu_request=0.45),
25
- service_account="tfy-workflows-sa",
26
- )
27
-
28
-
29
- # figure out naming
30
- @task(task_config=cpu_task_config)
31
- def should_train_tokenizer(tokenizer: str) -> bool:
32
- print("Should train tokenizer")
33
- return not bool(tokenizer)
34
-
35
-
36
- @task(
37
- task_config=cpu_task_config,
38
- # currently we will not support the caching, but keeping here for now
39
- # cache=True,
40
- # cache_version="1.0",
41
- )
42
- def train_tokenizer() -> str:
43
- print("Training tokenizer")
44
- return "trained_tokenizer"
45
-
46
-
47
- @task(
48
- task_config=PythonTaskConfig(
49
- image=TaskPythonBuild(
50
- python_version="3.9",
51
- pip_packages=["truefoundry[workflow]==0.4.1", "pynvml==11.5.0"],
52
- cuda_version="11.5-cudnn8",
53
- ),
54
- env={
55
- "NVIDIA_DRIVER_CAPABILITIES": "compute,utility",
56
- "NVIDIA_VISIBLE_DEVICES": "all",
57
- },
58
- resources=Resources(cpu_request=0.45, devices=[NvidiaGPU(name="T4", count=1)]),
59
- service_account="tfy-workflows-sa",
60
- ),
61
- )
62
- def train_model(tokenizer: str) -> Tuple[FlyteDirectory, str]:
63
- print("Training model")
64
- import flytekit
65
- # from pynvml import nvmlDeviceGetCount, nvmlInit
66
-
67
- # nvmlInit()
68
- # assert nvmlDeviceGetCount() > 0
69
-
70
- working_dir = flytekit.current_context().working_directory
71
- local_dir = Path(os.path.join(working_dir, "csv_files"))
72
- local_dir.mkdir(exist_ok=True)
73
-
74
- with open(os.path.join(local_dir, "model"), "w", encoding="utf-8") as f:
75
- f.write(tokenizer)
76
-
77
- return FlyteDirectory(path=str(local_dir)), "hello"
78
-
79
-
80
- @task(task_config=cpu_task_config)
81
- def get_validation_data() -> List[str]:
82
- print("Getting validation data")
83
- return ["foo", "bar", "baz"]
84
-
85
-
86
- @task(task_config=cpu_task_config)
87
- def validate_model(model: FlyteDirectory, tokenizer: str, validation_data: str) -> bool:
88
- print(validation_data)
89
- model_path = os.path.join(model, "model")
90
- with open(model_path, "r", encoding="utf-8") as f:
91
- return f.read() == tokenizer
92
-
93
-
94
- @task(task_config=cpu_task_config)
95
- def all_good(validations: List[bool]) -> bool:
96
- print("Validations", validations)
97
- return all(validations)
98
-
99
-
100
- echo = ContainerTask(
101
- name="echo",
102
- task_config=ContainerTaskConfig(
103
- image=Image(
104
- image_uri="docker.io/bash:4.1",
105
- command=["echo", "hello"],
106
- ),
107
- service_account="tfy-workflows-sa",
108
- ),
109
- )
110
-
111
-
112
- @task(task_config=cpu_task_config)
113
- def random(tokenizer: str) -> Tuple[FlyteDirectory, str]:
114
- print(tokenizer)
115
- return FlyteDirectory(path=""), "random"
116
-
117
-
118
- @workflow(
119
- # execution_configs=[
120
- # ExecutionConfig(
121
- # schedule="*/10 * * * *",
122
- # )
123
- # ]
124
- )
125
- def train(
126
- tokenizer: str = "",
127
- test_v2: str = "",
128
- optional_var: Optional[str] = "",
129
- default_v: Optional[str] = "hello1",
130
- default_v2: str = "hello2",
131
- ) -> bool:
132
- stt = should_train_tokenizer(tokenizer=tokenizer)
133
- model, t = (
134
- conditional("train_tokenizer")
135
- .if_(stt.is_true())
136
- .then(train_model(tokenizer=tokenizer))
137
- .else_()
138
- .then(random(tokenizer=tokenizer))
139
- )
140
- validation_task = partial(validate_model, model=model, tokenizer=t)
141
- validation_data = get_validation_data()
142
- validations = map_task(
143
- validation_task,
144
- concurrency=2,
145
- )(validation_data=validation_data)
146
- echo()
147
- return all_good(validations=validations)
148
-
149
-
150
- if __name__ == "__main__":
151
- train()
@@ -1,9 +0,0 @@
1
- type: workflow
2
- name: my-test-workflow9
3
- source:
4
- type: local
5
- local_build: false
6
- project_root_path: "./"
7
- workflow_file_path: package/test_workflow.py
8
- # workflow_file_path: hello_world_package/workflow.py
9
- flyte_entities: []