ob-metaflow-stubs 6.0.9.1__py2.py3-none-any.whl → 6.0.9.2__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.

Potentially problematic release.

This version of ob-metaflow-stubs might be problematic. Click here for more details.

Files changed (262) hide show
  1. metaflow-stubs/__init__.pyi +991 -991
  2. metaflow-stubs/cards.pyi +2 -2
  3. metaflow-stubs/cli.pyi +2 -2
  4. metaflow-stubs/cli_components/__init__.pyi +2 -2
  5. metaflow-stubs/cli_components/utils.pyi +2 -2
  6. metaflow-stubs/client/__init__.pyi +2 -2
  7. metaflow-stubs/client/core.pyi +4 -4
  8. metaflow-stubs/client/filecache.pyi +3 -3
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +3 -3
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +2 -2
  14. metaflow-stubs/meta_files.pyi +2 -2
  15. metaflow-stubs/metadata_provider/__init__.pyi +2 -2
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +2 -2
  17. metaflow-stubs/metadata_provider/metadata.pyi +2 -2
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +4 -2
  20. metaflow-stubs/metaflow_current.pyi +51 -51
  21. metaflow-stubs/metaflow_git.pyi +2 -2
  22. metaflow-stubs/mf_extensions/__init__.pyi +2 -2
  23. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +2 -2
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +2 -2
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +2 -2
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +2 -2
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +2 -2
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +2 -2
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +3 -3
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +2 -2
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +2 -2
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +3 -3
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +2 -2
  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 +2 -2
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +4 -4
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +4 -4
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +3 -3
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +2 -2
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +4 -4
  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 +2 -2
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
  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 +2 -2
  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 +2 -2
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +2 -2
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +3 -3
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +2 -2
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +2 -2
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +2 -2
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +3 -3
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +2 -2
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +2 -2
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +2 -2
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +3 -3
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +2 -2
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +2 -2
  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 +2 -2
  65. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +2 -2
  66. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/__init__.pyi +2 -2
  67. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/__init__.pyi +2 -2
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_state_machine.pyi +2 -2
  69. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_vendor/__init__.pyi +2 -2
  70. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_vendor/spinner/__init__.pyi +2 -2
  71. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_vendor/spinner/spinners.pyi +2 -2
  72. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/app_cli.pyi +3 -3
  73. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/app_config.pyi +3 -3
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/capsule.pyi +2 -2
  75. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/click_importer.pyi +2 -2
  76. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/code_package/__init__.pyi +2 -2
  77. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/code_package/code_packager.pyi +3 -3
  78. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/__init__.pyi +2 -2
  79. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/cli_generator.pyi +2 -2
  80. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/config_utils.pyi +3 -3
  81. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/schema_export.pyi +2 -2
  82. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/typed_configs.pyi +4 -4
  83. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/unified_config.pyi +3 -3
  84. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/dependencies.pyi +3 -3
  85. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/deployer.pyi +4 -4
  86. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/experimental/__init__.pyi +2 -2
  87. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/perimeters.pyi +2 -2
  88. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/utils.pyi +3 -3
  89. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/__init__.pyi +2 -2
  90. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/assume_role_decorator.pyi +3 -3
  91. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +2 -2
  92. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/async_cards.pyi +3 -3
  93. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +2 -2
  94. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +2 -2
  95. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +3 -3
  96. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +3 -3
  97. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/__init__.pyi +2 -2
  98. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/baker.pyi +3 -3
  99. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/docker_environment.pyi +3 -3
  100. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/fast_bakery.pyi +2 -2
  101. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/__init__.pyi +2 -2
  102. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/pod_killer.pyi +2 -2
  103. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +2 -2
  104. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +2 -2
  105. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +2 -2
  106. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +2 -2
  107. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/status_card.pyi +2 -2
  108. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +2 -2
  109. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +2 -2
  110. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
  111. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
  112. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +3 -3
  113. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
  114. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +2 -2
  115. metaflow-stubs/mf_extensions/outerbounds/toplevel/s3_proxy.pyi +2 -2
  116. metaflow-stubs/multicore_utils.pyi +2 -2
  117. metaflow-stubs/ob_internal.pyi +2 -2
  118. metaflow-stubs/packaging_sys/__init__.pyi +5 -5
  119. metaflow-stubs/packaging_sys/backend.pyi +4 -4
  120. metaflow-stubs/packaging_sys/distribution_support.pyi +4 -4
  121. metaflow-stubs/packaging_sys/tar_backend.pyi +5 -5
  122. metaflow-stubs/packaging_sys/utils.pyi +2 -2
  123. metaflow-stubs/packaging_sys/v1.pyi +2 -2
  124. metaflow-stubs/parameters.pyi +2 -2
  125. metaflow-stubs/plugins/__init__.pyi +11 -11
  126. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  127. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  128. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  129. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  130. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  131. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  132. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  133. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  134. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  135. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  136. metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -3
  137. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  138. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +3 -3
  139. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +3 -3
  140. metaflow-stubs/plugins/argo/exit_hooks.pyi +2 -2
  141. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  142. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  143. metaflow-stubs/plugins/aws/aws_utils.pyi +5 -2
  144. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  145. metaflow-stubs/plugins/aws/batch/batch.pyi +4 -4
  146. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  147. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +8 -2
  148. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  149. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +5 -5
  150. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  151. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  152. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  153. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +3 -3
  154. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  155. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +2 -2
  156. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +2 -2
  157. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  158. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  159. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  160. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +5 -5
  161. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  162. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  163. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  164. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  165. metaflow-stubs/plugins/cards/card_client.pyi +3 -3
  166. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  167. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  168. metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
  169. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  170. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  171. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  172. metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
  173. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  174. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  175. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  176. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  177. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  178. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  179. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  180. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  181. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  182. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  183. metaflow-stubs/plugins/datatools/s3/s3.pyi +4 -4
  184. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  185. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  186. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  187. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  188. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  189. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  190. metaflow-stubs/plugins/exit_hook/__init__.pyi +2 -2
  191. metaflow-stubs/plugins/exit_hook/exit_hook_decorator.pyi +2 -2
  192. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  193. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  194. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  195. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +5 -5
  196. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  197. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  198. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  199. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  200. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  201. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  202. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
  203. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  204. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
  205. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  206. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
  207. metaflow-stubs/plugins/ollama/__init__.pyi +3 -3
  208. metaflow-stubs/plugins/optuna/__init__.pyi +2 -2
  209. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  210. metaflow-stubs/plugins/perimeters.pyi +2 -2
  211. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  212. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  213. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  214. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  215. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  216. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  217. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  218. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  219. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  220. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  221. metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
  222. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +4 -4
  223. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  224. metaflow-stubs/plugins/secrets/secrets_func.pyi +2 -2
  225. metaflow-stubs/plugins/secrets/secrets_spec.pyi +2 -2
  226. metaflow-stubs/plugins/secrets/utils.pyi +2 -2
  227. metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
  228. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  229. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  230. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  231. metaflow-stubs/plugins/torchtune/__init__.pyi +2 -2
  232. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  233. metaflow-stubs/plugins/uv/uv_environment.pyi +3 -3
  234. metaflow-stubs/profilers/__init__.pyi +2 -2
  235. metaflow-stubs/pylint_wrapper.pyi +2 -2
  236. metaflow-stubs/runner/__init__.pyi +2 -2
  237. metaflow-stubs/runner/deployer.pyi +32 -32
  238. metaflow-stubs/runner/deployer_impl.pyi +2 -2
  239. metaflow-stubs/runner/metaflow_runner.pyi +3 -3
  240. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  241. metaflow-stubs/runner/nbrun.pyi +2 -2
  242. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  243. metaflow-stubs/runner/utils.pyi +3 -3
  244. metaflow-stubs/system/__init__.pyi +2 -2
  245. metaflow-stubs/system/system_logger.pyi +2 -2
  246. metaflow-stubs/system/system_monitor.pyi +2 -2
  247. metaflow-stubs/tagging_util.pyi +2 -2
  248. metaflow-stubs/tuple_util.pyi +2 -2
  249. metaflow-stubs/user_configs/__init__.pyi +2 -2
  250. metaflow-stubs/user_configs/config_options.pyi +2 -2
  251. metaflow-stubs/user_configs/config_parameters.pyi +6 -6
  252. metaflow-stubs/user_decorators/__init__.pyi +2 -2
  253. metaflow-stubs/user_decorators/common.pyi +2 -2
  254. metaflow-stubs/user_decorators/mutable_flow.pyi +5 -5
  255. metaflow-stubs/user_decorators/mutable_step.pyi +5 -5
  256. metaflow-stubs/user_decorators/user_flow_decorator.pyi +4 -4
  257. metaflow-stubs/user_decorators/user_step_decorator.pyi +5 -5
  258. {ob_metaflow_stubs-6.0.9.1.dist-info → ob_metaflow_stubs-6.0.9.2.dist-info}/METADATA +1 -1
  259. ob_metaflow_stubs-6.0.9.2.dist-info/RECORD +262 -0
  260. ob_metaflow_stubs-6.0.9.1.dist-info/RECORD +0 -262
  261. {ob_metaflow_stubs-6.0.9.1.dist-info → ob_metaflow_stubs-6.0.9.2.dist-info}/WHEEL +0 -0
  262. {ob_metaflow_stubs-6.0.9.1.dist-info → ob_metaflow_stubs-6.0.9.2.dist-info}/top_level.txt +0 -0
metaflow-stubs/packaging_sys/backend.pyi CHANGED
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.095041 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.346518 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import abc
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
12
  import abc
14
13
  import io
14
+ import typing
15
15
 
16
16
 
17
17
  class PackagingBackend(abc.ABC, metaclass=abc.ABCMeta):
metaflow-stubs/packaging_sys/distribution_support.pyi CHANGED
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.093618 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.345655 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
+ import importlib
9
10
  import typing
10
11
  import abc
11
- import importlib
12
12
  if typing.TYPE_CHECKING:
13
- import abc
14
13
  import os
15
14
  import importlib.metadata
15
+ import abc
16
16
 
17
17
 
18
18
  TYPE_CHECKING: bool
metaflow-stubs/packaging_sys/tar_backend.pyi CHANGED
@@ -1,19 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.095659 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.347115 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
+ import abc
9
10
  import typing
10
11
  import metaflow
11
- import abc
12
12
  if typing.TYPE_CHECKING:
13
- import typing
14
13
  import metaflow.packaging_sys.backend
15
- import abc
16
14
  import io
15
+ import typing
16
+ import abc
17
17
  import tarfile
18
18
 
19
19
  from .backend import PackagingBackend as PackagingBackend
metaflow-stubs/packaging_sys/utils.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.095991 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.347469 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/packaging_sys/v1.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.096743 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.348212 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/parameters.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.065001 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.315699 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.072931 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.323920 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,25 +9,25 @@ from __future__ import annotations
9
9
 
10
10
  from . import test_unbounded_foreach_decorator as test_unbounded_foreach_decorator
11
11
  from .test_unbounded_foreach_decorator import InternalTestUnboundedForeachInput as InternalTestUnboundedForeachInput
12
+ from . import catch_decorator as catch_decorator
12
13
  from . import pypi as pypi
13
- from . import cards as cards
14
- from . import environment_decorator as environment_decorator
15
14
  from . import datatools as datatools
16
15
  from . import gcp as gcp
17
16
  from . import storage_executor as storage_executor
17
+ from . import cards as cards
18
18
  from . import secrets as secrets
19
- from . import retry_decorator as retry_decorator
19
+ from . import frameworks as frameworks
20
+ from . import parallel_decorator as parallel_decorator
21
+ from . import timeout_decorator as timeout_decorator
22
+ from . import argo as argo
23
+ from . import environment_decorator as environment_decorator
20
24
  from . import kubernetes as kubernetes
21
25
  from . import resources_decorator as resources_decorator
22
- from . import timeout_decorator as timeout_decorator
23
26
  from . import aws as aws
24
- from . import argo as argo
25
- from . import parallel_decorator as parallel_decorator
26
- from . import catch_decorator as catch_decorator
27
- from . import frameworks as frameworks
28
- from . import events_decorator as events_decorator
27
+ from . import retry_decorator as retry_decorator
29
28
  from . import project_decorator as project_decorator
30
29
  from . import airflow as airflow
30
+ from . import events_decorator as events_decorator
31
31
  from . import exit_hook as exit_hook
32
32
  from . import uv as uv
33
33
  from . import azure as azure
metaflow-stubs/plugins/airflow/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.113925 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.365366 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/airflow/airflow_utils.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.148918 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.401325 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/airflow/exception.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.148039 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.400417 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/airflow/sensors/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.149173 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.401587 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.246603 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.499706 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.246877 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.499976 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.247140 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.500211 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/argo/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.111848 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.363784 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/argo/argo_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.259756 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.503303 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/argo/argo_events.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.146826 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.396020 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/argo/argo_workflows.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.264064 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.507589 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.exception
13
12
  import metaflow.graph
13
+ import metaflow.exception
14
14
 
15
15
  from ...parameters import JSONType as JSONType
16
16
  from ...metaflow_current import current as current
metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.147222 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.396300 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
+ import metaflow.decorators
12
13
  import metaflow.flowspec
13
14
  import metaflow.graph
14
- import metaflow.decorators
15
15
 
16
16
  from ...metaflow_current import current as current
17
17
  from ...events import Trigger as Trigger
metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.147609 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.396802 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
+ import metaflow.plugins.argo.argo_workflows_deployer_objects
12
13
  import metaflow.runner.deployer_impl
13
14
  import typing
14
- import metaflow.plugins.argo.argo_workflows_deployer_objects
15
15
 
16
16
  from ...runner.deployer_impl import DeployerImpl as DeployerImpl
17
17
 
metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.246234 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.496448 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.runner.deployer
13
12
  import metaflow.plugins.argo.argo_workflows_deployer_objects
13
+ import metaflow.runner.deployer
14
14
 
15
15
  from ...client.core import get_metadata as get_metadata
16
16
  from ...exception import MetaflowException as MetaflowException
metaflow-stubs/plugins/argo/exit_hooks.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.267595 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.519129 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.111616 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.364539 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/aws_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.146227 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.399981 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/aws_utils.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.145537 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.399266 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -89,3 +89,6 @@ def sanitize_batch_tag(key, value):
89
89
  """
90
90
  ...
91
91
 
92
+ def validate_aws_tag(key: str, value: str):
93
+ ...
94
+
metaflow-stubs/plugins/aws/batch/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.145952 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.399737 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/batch/batch.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.168089 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.499046 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -65,9 +65,9 @@ class Batch(object, metaclass=type):
65
65
  ...
66
66
  def kill_jobs(self, flow_name, run_id, user, echo):
67
67
  ...
68
- def create_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, env = {}, attrs = {}, host_volumes = None, efs_volumes = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, ephemeral_storage = None, log_driver = None, log_options = None):
68
+ def create_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, env = {}, attrs = {}, host_volumes = None, efs_volumes = None, use_tmpfs = None, aws_batch_tags = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, ephemeral_storage = None, log_driver = None, log_options = None):
69
69
  ...
70
- def launch_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, host_volumes = None, efs_volumes = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, env = {}, attrs = {}, ephemeral_storage = None, log_driver = None, log_options = None):
70
+ def launch_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, host_volumes = None, efs_volumes = None, use_tmpfs = None, aws_batch_tags = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, env = {}, attrs = {}, ephemeral_storage = None, log_driver = None, log_options = None):
71
71
  ...
72
72
  def wait(self, stdout_location, stderr_location, echo = None):
73
73
  ...
metaflow-stubs/plugins/aws/batch/batch_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.167424 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.498358 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/batch/batch_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.162776 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.415401 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -18,12 +18,15 @@ from ...timeout_decorator import get_run_time_limit_for_task as get_run_time_lim
18
18
  from ..aws_utils import compute_resource_attributes as compute_resource_attributes
19
19
  from ..aws_utils import get_docker_registry as get_docker_registry
20
20
  from ..aws_utils import get_ec2_instance_metadata as get_ec2_instance_metadata
21
+ from ..aws_utils import validate_aws_tag as validate_aws_tag
21
22
  from .batch import BatchException as BatchException
22
23
 
23
24
  BATCH_CONTAINER_IMAGE: None
24
25
 
25
26
  BATCH_CONTAINER_REGISTRY: None
26
27
 
28
+ BATCH_DEFAULT_TAGS: dict
29
+
27
30
  BATCH_JOB_QUEUE: None
28
31
 
29
32
  DATASTORE_LOCAL_DIR: str
@@ -76,6 +79,9 @@ class BatchDecorator(metaflow.decorators.StepDecorator, metaclass=type):
76
79
  A swappiness value of 0 causes swapping not to happen unless absolutely
77
80
  necessary. A swappiness value of 100 causes pages to be swapped very
78
81
  aggressively. Accepted values are whole numbers between 0 and 100.
82
+ aws_batch_tags: Dict[str, str], optional, default None
83
+ Sets arbitrary AWS tags on the AWS Batch compute environment.
84
+ Set as string key-value pairs.
79
85
  use_tmpfs : bool, default False
80
86
  This enables an explicit tmpfs mount for this step. Note that tmpfs is
81
87
  not available on Fargate compute environments
metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.146428 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.400196 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi CHANGED
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.168387 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.499367 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import abc
10
+ import metaflow
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
- import metaflow.exception
14
- import metaflow.plugins.secrets
15
13
  import abc
14
+ import metaflow.plugins.secrets
15
+ import metaflow.exception
16
16
 
17
17
  from ....exception import MetaflowException as MetaflowException
18
18
  from ...secrets import SecretsProvider as SecretsProvider
metaflow-stubs/plugins/aws/step_functions/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.145734 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.399490 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.267879 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.520926 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.165625 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.496692 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/step_functions.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.265727 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.520626 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -37,7 +37,7 @@ class StepFunctionsSchedulingException(metaflow.exception.MetaflowException, met
37
37
  ...
38
38
 
39
39
  class StepFunctions(object, metaclass=type):
40
- def __init__(self, name, graph, flow, code_package_metadata, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, is_project = False, use_distributed_map = False):
40
+ def __init__(self, name, graph, flow, code_package_metadata, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, aws_batch_tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, is_project = False, use_distributed_map = False):
41
41
  ...
42
42
  def to_json(self):
43
43
  ...
metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.268227 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.521256 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.0.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-08-28T00:53:38.166169 #
3
+ # MF version: 2.18.1.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-29T18:32:22.497105 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations