ob-metaflow-stubs 6.0.3.171__py2.py3-none-any.whl → 6.0.3.173rc0__py2.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.
Files changed (214) hide show
  1. metaflow-stubs/__init__.pyi +708 -707
  2. metaflow-stubs/cards.pyi +1 -1
  3. metaflow-stubs/cli.pyi +1 -1
  4. metaflow-stubs/cli_components/__init__.pyi +1 -1
  5. metaflow-stubs/cli_components/utils.pyi +1 -1
  6. metaflow-stubs/client/__init__.pyi +1 -1
  7. metaflow-stubs/client/core.pyi +4 -4
  8. metaflow-stubs/client/filecache.pyi +1 -1
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +1 -1
  11. metaflow-stubs/flowspec.pyi +4 -4
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +3 -3
  14. metaflow-stubs/info_file.pyi +1 -1
  15. metaflow-stubs/metadata_provider/__init__.pyi +1 -1
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +1 -1
  17. metaflow-stubs/metadata_provider/metadata.pyi +2 -2
  18. metaflow-stubs/metadata_provider/util.pyi +1 -1
  19. metaflow-stubs/metaflow_config.pyi +1 -1
  20. metaflow-stubs/metaflow_current.pyi +123 -123
  21. metaflow-stubs/metaflow_git.pyi +1 -1
  22. metaflow-stubs/mf_extensions/__init__.pyi +1 -1
  23. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +1 -1
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +1 -1
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +1 -1
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +1 -1
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +1 -1
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +1 -1
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +2 -2
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +1 -1
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +1 -1
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +4 -4
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +1 -1
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +4 -4
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +1 -1
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +3 -3
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +3 -3
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +1 -1
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +1 -1
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +1 -1
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +1 -1
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +2 -2
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +1 -1
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +1 -1
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +3 -3
  47. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +1 -1
  48. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +2 -2
  49. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +1 -1
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +1 -1
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +1 -1
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +1 -1
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +1 -1
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +1 -1
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +4 -4
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +1 -1
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +1 -1
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +1 -1
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +2 -2
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +1 -1
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +1 -1
  62. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +2 -2
  63. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +2 -2
  64. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +1 -1
  65. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +1 -1
  66. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +1 -1
  67. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +1 -1
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +1 -1
  69. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +1 -1
  70. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +1 -1
  71. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/__init__.pyi +6 -0
  72. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/baker.pyi +51 -0
  73. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/docker_environment.pyi +65 -0
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/fast_bakery.pyi +74 -0
  75. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +1 -1
  76. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +1 -1
  77. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +1 -1
  78. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +1 -1
  79. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +1 -1
  80. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +1 -1
  81. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +1 -1
  82. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +1 -1
  83. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +1 -1
  84. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +1 -1
  85. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +2 -1
  86. metaflow-stubs/multicore_utils.pyi +1 -1
  87. metaflow-stubs/ob_internal.pyi +11 -0
  88. metaflow-stubs/parameters.pyi +3 -3
  89. metaflow-stubs/plugins/__init__.pyi +12 -11
  90. metaflow-stubs/plugins/airflow/__init__.pyi +1 -1
  91. metaflow-stubs/plugins/airflow/airflow_utils.pyi +1 -1
  92. metaflow-stubs/plugins/airflow/exception.pyi +1 -1
  93. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +1 -1
  94. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +1 -1
  95. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +1 -1
  96. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +1 -1
  97. metaflow-stubs/plugins/argo/__init__.pyi +1 -1
  98. metaflow-stubs/plugins/argo/argo_client.pyi +1 -1
  99. metaflow-stubs/plugins/argo/argo_events.pyi +1 -1
  100. metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -3
  101. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +2 -2
  102. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +3 -3
  103. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +2 -2
  104. metaflow-stubs/plugins/aws/__init__.pyi +1 -1
  105. metaflow-stubs/plugins/aws/aws_client.pyi +1 -1
  106. metaflow-stubs/plugins/aws/aws_utils.pyi +1 -1
  107. metaflow-stubs/plugins/aws/batch/__init__.pyi +1 -1
  108. metaflow-stubs/plugins/aws/batch/batch.pyi +1 -1
  109. metaflow-stubs/plugins/aws/batch/batch_client.pyi +1 -1
  110. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +1 -1
  111. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +1 -1
  112. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +2 -2
  113. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +1 -1
  114. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +1 -1
  115. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +1 -1
  116. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +1 -1
  117. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +1 -1
  118. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +3 -3
  119. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +2 -2
  120. metaflow-stubs/plugins/azure/__init__.pyi +1 -1
  121. metaflow-stubs/plugins/azure/azure_credential.pyi +1 -1
  122. metaflow-stubs/plugins/azure/azure_exceptions.pyi +1 -1
  123. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +2 -2
  124. metaflow-stubs/plugins/azure/azure_utils.pyi +1 -1
  125. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +1 -1
  126. metaflow-stubs/plugins/azure/includefile_support.pyi +1 -1
  127. metaflow-stubs/plugins/cards/__init__.pyi +5 -5
  128. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  129. metaflow-stubs/plugins/cards/card_creator.pyi +1 -1
  130. metaflow-stubs/plugins/cards/card_datastore.pyi +1 -1
  131. metaflow-stubs/plugins/cards/card_decorator.pyi +1 -1
  132. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +1 -1
  133. metaflow-stubs/plugins/cards/card_modules/basic.pyi +1 -1
  134. metaflow-stubs/plugins/cards/card_modules/card.pyi +1 -1
  135. metaflow-stubs/plugins/cards/card_modules/components.pyi +3 -3
  136. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +1 -1
  137. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +1 -1
  138. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +1 -1
  139. metaflow-stubs/plugins/cards/card_resolver.pyi +1 -1
  140. metaflow-stubs/plugins/cards/component_serializer.pyi +1 -1
  141. metaflow-stubs/plugins/cards/exception.pyi +1 -1
  142. metaflow-stubs/plugins/catch_decorator.pyi +1 -1
  143. metaflow-stubs/plugins/datatools/__init__.pyi +1 -1
  144. metaflow-stubs/plugins/datatools/local.pyi +1 -1
  145. metaflow-stubs/plugins/datatools/s3/__init__.pyi +1 -1
  146. metaflow-stubs/plugins/datatools/s3/s3.pyi +3 -3
  147. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +1 -1
  148. metaflow-stubs/plugins/datatools/s3/s3util.pyi +1 -1
  149. metaflow-stubs/plugins/debug_logger.pyi +1 -1
  150. metaflow-stubs/plugins/debug_monitor.pyi +1 -1
  151. metaflow-stubs/plugins/environment_decorator.pyi +1 -1
  152. metaflow-stubs/plugins/events_decorator.pyi +1 -1
  153. metaflow-stubs/plugins/frameworks/__init__.pyi +1 -1
  154. metaflow-stubs/plugins/frameworks/pytorch.pyi +1 -1
  155. metaflow-stubs/plugins/gcp/__init__.pyi +1 -1
  156. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +2 -2
  157. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +1 -1
  158. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +1 -1
  159. metaflow-stubs/plugins/gcp/gs_utils.pyi +1 -1
  160. metaflow-stubs/plugins/gcp/includefile_support.pyi +1 -1
  161. metaflow-stubs/plugins/kubernetes/__init__.pyi +1 -1
  162. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  163. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +1 -1
  164. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +1 -1
  165. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +1 -1
  166. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +1 -1
  167. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +1 -1
  168. metaflow-stubs/plugins/ollama/__init__.pyi +2 -2
  169. metaflow-stubs/plugins/parallel_decorator.pyi +1 -1
  170. metaflow-stubs/plugins/perimeters.pyi +1 -1
  171. metaflow-stubs/plugins/project_decorator.pyi +1 -1
  172. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  173. metaflow-stubs/plugins/pypi/conda_decorator.pyi +1 -1
  174. metaflow-stubs/plugins/pypi/conda_environment.pyi +2 -2
  175. metaflow-stubs/plugins/pypi/parsers.pyi +1 -1
  176. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +1 -1
  177. metaflow-stubs/plugins/pypi/pypi_environment.pyi +1 -1
  178. metaflow-stubs/plugins/pypi/utils.pyi +1 -1
  179. metaflow-stubs/plugins/resources_decorator.pyi +1 -1
  180. metaflow-stubs/plugins/retry_decorator.pyi +1 -1
  181. metaflow-stubs/plugins/secrets/__init__.pyi +1 -1
  182. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +1 -1
  183. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +1 -1
  184. metaflow-stubs/plugins/snowflake/__init__.pyi +1 -1
  185. metaflow-stubs/plugins/storage_executor.pyi +1 -1
  186. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  187. metaflow-stubs/plugins/timeout_decorator.pyi +1 -1
  188. metaflow-stubs/plugins/torchtune/__init__.pyi +39 -0
  189. metaflow-stubs/plugins/uv/__init__.pyi +1 -1
  190. metaflow-stubs/plugins/uv/uv_environment.pyi +2 -2
  191. metaflow-stubs/profilers/__init__.pyi +1 -1
  192. metaflow-stubs/pylint_wrapper.pyi +1 -1
  193. metaflow-stubs/runner/__init__.pyi +1 -1
  194. metaflow-stubs/runner/deployer.pyi +27 -27
  195. metaflow-stubs/runner/deployer_impl.pyi +1 -1
  196. metaflow-stubs/runner/metaflow_runner.pyi +2 -2
  197. metaflow-stubs/runner/nbdeploy.pyi +1 -1
  198. metaflow-stubs/runner/nbrun.pyi +1 -1
  199. metaflow-stubs/runner/subprocess_manager.pyi +1 -1
  200. metaflow-stubs/runner/utils.pyi +3 -3
  201. metaflow-stubs/system/__init__.pyi +1 -1
  202. metaflow-stubs/system/system_logger.pyi +1 -1
  203. metaflow-stubs/system/system_monitor.pyi +1 -1
  204. metaflow-stubs/tagging_util.pyi +1 -1
  205. metaflow-stubs/tuple_util.pyi +1 -1
  206. metaflow-stubs/user_configs/__init__.pyi +1 -1
  207. metaflow-stubs/user_configs/config_decorators.pyi +4 -4
  208. metaflow-stubs/user_configs/config_options.pyi +3 -3
  209. metaflow-stubs/user_configs/config_parameters.pyi +6 -6
  210. {ob_metaflow_stubs-6.0.3.171.dist-info → ob_metaflow_stubs-6.0.3.173rc0.dist-info}/METADATA +1 -1
  211. ob_metaflow_stubs-6.0.3.173rc0.dist-info/RECORD +214 -0
  212. ob_metaflow_stubs-6.0.3.171.dist-info/RECORD +0 -208
  213. {ob_metaflow_stubs-6.0.3.171.dist-info → ob_metaflow_stubs-6.0.3.173rc0.dist-info}/WHEEL +0 -0
  214. {ob_metaflow_stubs-6.0.3.171.dist-info → ob_metaflow_stubs-6.0.3.173rc0.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.135395 #
4
+ # Generated on 2025-05-28T01:05:50.909114 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.160356 #
4
+ # Generated on 2025-05-28T01:05:50.941964 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.exception
13
13
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.160017 #
4
+ # Generated on 2025-05-28T01:05:50.941628 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.092664 #
4
+ # Generated on 2025-05-28T01:05:50.835460 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.161070 #
4
+ # Generated on 2025-05-28T01:05:50.942688 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.224893 #
4
+ # Generated on 2025-05-28T01:05:51.009291 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.161299 #
4
+ # Generated on 2025-05-28T01:05:50.942913 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.093609 #
4
+ # Generated on 2025-05-28T01:05:50.836601 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.outerbounds.plugins.card_utilities.injector
13
12
  import metaflow.decorators
13
+ import metaflow.mf_extensions.outerbounds.plugins.card_utilities.injector
14
14
 
15
15
  from ...metaflow_current import current as current
16
16
  from ...mf_extensions.outerbounds.plugins.ollama import constants as constants
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.134476 #
4
+ # Generated on 2025-05-28T01:05:50.904800 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.093928 #
4
+ # Generated on 2025-05-28T01:05:50.839463 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.136022 #
4
+ # Generated on 2025-05-28T01:05:50.910631 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.134822 #
4
+ # Generated on 2025-05-28T01:05:50.906473 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
 
10
10
  from ... import metaflow_config as metaflow_config
11
11
  from ...exception import MetaflowException as MetaflowException
12
- from . import pypi_decorator as pypi_decorator
13
12
  from . import conda_decorator as conda_decorator
13
+ from . import pypi_decorator as pypi_decorator
14
14
  from . import utils as utils
15
15
  from . import conda_environment as conda_environment
16
16
  from . import pypi_environment as pypi_environment
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.158113 #
4
+ # Generated on 2025-05-28T01:05:50.932169 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.159075 #
4
+ # Generated on 2025-05-28T01:05:50.933582 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -12,9 +12,9 @@ import io
12
12
  import typing
13
13
  if typing.TYPE_CHECKING:
14
14
  import abc
15
+ import metaflow.metaflow_environment
15
16
  import metaflow.exception
16
17
  import io
17
- import metaflow.metaflow_environment
18
18
 
19
19
  from ...exception import MetaflowException as MetaflowException
20
20
  from ...metaflow_config import get_pinned_conda_libs as get_pinned_conda_libs
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.106063 #
4
+ # Generated on 2025-05-28T01:05:50.860231 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.157523 #
4
+ # Generated on 2025-05-28T01:05:50.932521 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.159209 #
4
+ # Generated on 2025-05-28T01:05:50.933715 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.158352 #
4
+ # Generated on 2025-05-28T01:05:50.932756 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.131356 #
4
+ # Generated on 2025-05-28T01:05:50.908537 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.130628 #
4
+ # Generated on 2025-05-28T01:05:50.908862 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.130930 #
4
+ # Generated on 2025-05-28T01:05:50.906229 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.147539 #
4
+ # Generated on 2025-05-28T01:05:50.931261 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.147361 #
4
+ # Generated on 2025-05-28T01:05:50.931083 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.093085 #
4
+ # Generated on 2025-05-28T01:05:50.835861 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.132627 #
4
+ # Generated on 2025-05-28T01:05:50.908334 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.130408 #
4
+ # Generated on 2025-05-28T01:05:50.904170 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.decorators
13
12
  import metaflow.unbounded_foreach
13
+ import metaflow.decorators
14
14
 
15
15
  from ..exception import MetaflowException as MetaflowException
16
16
  from ..metadata_provider.metadata import MetaDatum as MetaDatum
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.131731 #
4
+ # Generated on 2025-05-28T01:05:50.905325 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -0,0 +1,39 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
+ # Generated on 2025-05-28T01:05:50.839011 #
5
+ ######################################################################################################
6
+
7
+ from __future__ import annotations
8
+
9
+ import typing
10
+
11
+ from ...metaflow_current import current as current
12
+
13
+ class TorchTune(object, metaclass=type):
14
+ def __init__(self, use_multi_node_config: bool = False):
15
+ """
16
+ Initialize the Tune launcher.
17
+
18
+ :param use_multi_node_config: If True, attempt to build a distributed configuration
19
+ from current.torch.torchrun_args.
20
+ """
21
+ ...
22
+ def run(self, recipe: str, config_dict: typing.Dict, additional_cli_options: typing.Optional[typing.List[str]] = None):
23
+ """
24
+ Launch the torchtune job via its CLI.
25
+
26
+ :param recipe: The path to the recipe (or name of the recipe) to run.
27
+ :param config_dict: Optional dictionary that will be dumped to a YAML file and passed via --config.
28
+ :param additional_cli_options: Optional list of additional CLI options.
29
+ :raises: subprocess.CalledProcessError if the subprocess returns a nonzero exit code.
30
+ """
31
+ ...
32
+ ...
33
+
34
+ def enqueue_output(file, queue):
35
+ ...
36
+
37
+ def read_popen_pipes(p):
38
+ ...
39
+
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.136659 #
4
+ # Generated on 2025-05-28T01:05:50.912069 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.162979 #
4
+ # Generated on 2025-05-28T01:05:50.944928 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.exception
13
12
  import metaflow.metaflow_environment
13
+ import metaflow.exception
14
14
 
15
15
  from ...exception import MetaflowException as MetaflowException
16
16
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.094172 #
4
+ # Generated on 2025-05-28T01:05:50.839789 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.117596 #
4
+ # Generated on 2025-05-28T01:05:50.880438 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.103483 #
4
+ # Generated on 2025-05-28T01:05:50.855738 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.237634 #
4
+ # Generated on 2025-05-28T01:05:51.022518 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
13
- import metaflow.plugins.argo.argo_workflows_deployer_objects
14
13
  import metaflow
15
14
  import metaflow.plugins.aws.step_functions.step_functions_deployer
15
+ import metaflow.plugins.argo.argo_workflows_deployer_objects
16
16
  import metaflow.plugins.argo.argo_workflows_deployer
17
17
  import metaflow.runner.deployer
18
18
 
@@ -55,24 +55,24 @@ class Deployer(object, metaclass=DeployerMeta):
55
55
  """
56
56
  def __init__(self, flow_file: str, show_output: bool = True, profile: typing.Optional[str] = None, env: typing.Optional[typing.Dict] = None, cwd: typing.Optional[str] = None, file_read_timeout: int = 3600, **kwargs):
57
57
  ...
58
- def step_functions(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.aws.step_functions.step_functions_deployer.StepFunctionsDeployer":
58
+ def argo_workflows(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer.ArgoWorkflowsDeployer":
59
59
  """
60
- Deployer implementation for AWS Step Functions.
60
+ Deployer implementation for Argo Workflows.
61
61
 
62
62
  Parameters
63
63
  ----------
64
64
  name : str, optional, default None
65
- State Machine name. The flow name is used instead if this option is not specified.
65
+ Argo workflow name. The flow name is used instead if this option is not specified.
66
66
  """
67
67
  ...
68
- def argo_workflows(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer.ArgoWorkflowsDeployer":
68
+ def step_functions(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.aws.step_functions.step_functions_deployer.StepFunctionsDeployer":
69
69
  """
70
- Deployer implementation for Argo Workflows.
70
+ Deployer implementation for AWS Step Functions.
71
71
 
72
72
  Parameters
73
73
  ----------
74
74
  name : str, optional, default None
75
- Argo workflow name. The flow name is used instead if this option is not specified.
75
+ State Machine name. The flow name is used instead if this option is not specified.
76
76
  """
77
77
  ...
78
78
  ...
@@ -159,24 +159,6 @@ class DeployedFlow(object, metaclass=DeployedFlowMeta):
159
159
  """
160
160
  ...
161
161
  @classmethod
162
- def from_step_functions(cls) -> "metaflow.plugins.aws.step_functions.step_functions_deployer_objects.StepFunctionsDeployedFlow":
163
- """
164
- This method is not currently implemented for Step Functions.
165
-
166
- Raises
167
- ------
168
- NotImplementedError
169
- This method is not implemented for Step Functions.
170
-
171
- Parameters
172
- ----------
173
-
174
-
175
- Returns
176
- -------
177
- """
178
- ...
179
- @classmethod
180
162
  def from_argo_workflows(cls, *, identifier: str, metadata: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer_objects.ArgoWorkflowsDeployedFlow":
181
163
  """
182
164
  Retrieves a `ArgoWorkflowsDeployedFlow` object from an identifier and optional
@@ -200,5 +182,23 @@ class DeployedFlow(object, metaclass=DeployedFlowMeta):
200
182
  deployed flow on argo workflows.
201
183
  """
202
184
  ...
185
+ @classmethod
186
+ def from_step_functions(cls) -> "metaflow.plugins.aws.step_functions.step_functions_deployer_objects.StepFunctionsDeployedFlow":
187
+ """
188
+ This method is not currently implemented for Step Functions.
189
+
190
+ Raises
191
+ ------
192
+ NotImplementedError
193
+ This method is not implemented for Step Functions.
194
+
195
+ Parameters
196
+ ----------
197
+
198
+
199
+ Returns
200
+ -------
201
+ """
202
+ ...
203
203
  ...
204
204
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.129900 #
4
+ # Generated on 2025-05-28T01:05:50.903130 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.111894 #
4
+ # Generated on 2025-05-28T01:05:50.870921 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.client.core
13
13
  import metaflow.runner.metaflow_runner
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.112857 #
4
+ # Generated on 2025-05-28T01:05:50.872326 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.112402 #
4
+ # Generated on 2025-05-28T01:05:50.871762 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.129373 #
4
+ # Generated on 2025-05-28T01:05:50.902187 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,16 +1,16 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.128349 #
4
+ # Generated on 2025-05-28T01:05:50.900204 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import metaflow.runner.click_api
12
- import typing
13
11
  import metaflow.runner.subprocess_manager
12
+ import typing
13
+ import metaflow.runner.click_api
14
14
 
15
15
 
16
16
  TYPE_CHECKING: bool
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-27T00:40:42.117288 #
4
+ # Generated on 2025-05-28T01:05:50.879967 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations