ob-metaflow-stubs 6.0.4.9__py2.py3-none-any.whl → 6.0.5.1__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 +1094 -1019
  2. metaflow-stubs/cards.pyi +2 -2
  3. metaflow-stubs/cli.pyi +3 -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 +7 -8
  8. metaflow-stubs/client/filecache.pyi +2 -2
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +8 -8
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +4 -4
  14. metaflow-stubs/{info_file.pyi → meta_files.pyi} +2 -6
  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 +3 -3
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +6 -2
  20. metaflow-stubs/metaflow_current.pyi +49 -49
  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 +3 -3
  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 +4 -4
  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 +4 -4
  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 +5 -5
  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 +2 -2
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +3 -3
  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 +3 -3
  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 +3 -3
  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 +4 -4
  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 +2 -2
  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 +3 -3
  63. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +3 -3
  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 +6 -6
  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 +2 -2
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/capsule.pyi +7 -7
  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 +2 -2
  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 +2 -2
  85. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/deployer.pyi +5 -5
  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 +6 -6
  89. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/__init__.pyi +2 -2
  90. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/assume_role_decorator.pyi +12 -8
  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 +13 -8
  96. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +11 -8
  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 +2 -2
  113. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
  114. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +13 -2
  115. metaflow-stubs/mf_extensions/outerbounds/toplevel/s3_proxy.pyi +24 -0
  116. metaflow-stubs/multicore_utils.pyi +2 -2
  117. metaflow-stubs/ob_internal.pyi +2 -2
  118. metaflow-stubs/packaging_sys/__init__.pyi +430 -0
  119. metaflow-stubs/packaging_sys/backend.pyi +86 -0
  120. metaflow-stubs/packaging_sys/distribution_support.pyi +57 -0
  121. metaflow-stubs/packaging_sys/tar_backend.pyi +62 -0
  122. metaflow-stubs/packaging_sys/utils.pyi +26 -0
  123. metaflow-stubs/packaging_sys/v1.pyi +145 -0
  124. metaflow-stubs/parameters.pyi +4 -4
  125. metaflow-stubs/plugins/__init__.pyi +14 -14
  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 +4 -4
  137. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +2 -2
  138. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  139. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +3 -3
  140. metaflow-stubs/plugins/argo/exit_hooks.pyi +3 -3
  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 +2 -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 +4 -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 +3 -3
  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 +4 -4
  156. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
  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 +3 -3
  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 +3 -5
  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 +2 -2
  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 +3 -3
  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 +3 -3
  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 +3 -3
  202. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +4 -4
  203. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  204. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +5 -3
  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 +2 -2
  208. metaflow-stubs/plugins/parallel_decorator.pyi +3 -3
  209. metaflow-stubs/plugins/perimeters.pyi +2 -2
  210. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  211. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  212. metaflow-stubs/plugins/pypi/conda_decorator.pyi +5 -8
  213. metaflow-stubs/plugins/pypi/conda_environment.pyi +4 -3
  214. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  215. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +4 -4
  216. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  217. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  218. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  219. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  220. metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
  221. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +2 -2
  222. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  223. metaflow-stubs/plugins/secrets/secrets_func.pyi +2 -2
  224. metaflow-stubs/plugins/secrets/secrets_spec.pyi +2 -2
  225. metaflow-stubs/plugins/secrets/utils.pyi +2 -2
  226. metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
  227. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  228. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
  229. metaflow-stubs/plugins/timeout_decorator.pyi +2 -2
  230. metaflow-stubs/plugins/torchtune/__init__.pyi +2 -2
  231. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  232. metaflow-stubs/plugins/uv/uv_environment.pyi +3 -2
  233. metaflow-stubs/profilers/__init__.pyi +2 -2
  234. metaflow-stubs/pylint_wrapper.pyi +2 -2
  235. metaflow-stubs/runner/__init__.pyi +2 -2
  236. metaflow-stubs/runner/deployer.pyi +34 -34
  237. metaflow-stubs/runner/deployer_impl.pyi +3 -3
  238. metaflow-stubs/runner/metaflow_runner.pyi +4 -4
  239. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  240. metaflow-stubs/runner/nbrun.pyi +2 -2
  241. metaflow-stubs/runner/subprocess_manager.pyi +3 -2
  242. metaflow-stubs/runner/utils.pyi +3 -3
  243. metaflow-stubs/system/__init__.pyi +2 -2
  244. metaflow-stubs/system/system_logger.pyi +3 -3
  245. metaflow-stubs/system/system_monitor.pyi +2 -2
  246. metaflow-stubs/tagging_util.pyi +2 -2
  247. metaflow-stubs/tuple_util.pyi +2 -2
  248. metaflow-stubs/user_configs/__init__.pyi +2 -3
  249. metaflow-stubs/user_configs/config_options.pyi +4 -5
  250. metaflow-stubs/user_configs/config_parameters.pyi +6 -8
  251. metaflow-stubs/user_decorators/__init__.pyi +15 -0
  252. metaflow-stubs/user_decorators/common.pyi +38 -0
  253. metaflow-stubs/user_decorators/mutable_flow.pyi +223 -0
  254. metaflow-stubs/user_decorators/mutable_step.pyi +152 -0
  255. metaflow-stubs/user_decorators/user_flow_decorator.pyi +137 -0
  256. metaflow-stubs/user_decorators/user_step_decorator.pyi +323 -0
  257. {ob_metaflow_stubs-6.0.4.9.dist-info → ob_metaflow_stubs-6.0.5.1.dist-info}/METADATA +1 -1
  258. ob_metaflow_stubs-6.0.5.1.dist-info/RECORD +261 -0
  259. metaflow-stubs/user_configs/config_decorators.pyi +0 -251
  260. ob_metaflow_stubs-6.0.4.9.dist-info/RECORD +0 -249
  261. {ob_metaflow_stubs-6.0.4.9.dist-info → ob_metaflow_stubs-6.0.5.1.dist-info}/WHEEL +0 -0
  262. {ob_metaflow_stubs-6.0.4.9.dist-info → ob_metaflow_stubs-6.0.5.1.dist-info}/top_level.txt +0 -0
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/app_config.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.296767 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.716106 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/capsule.pyi CHANGED
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.295594 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.714909 #
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.mf_extensions.outerbounds.plugins.apps.core.capsule
12
11
  import typing
13
12
  import metaflow.mf_extensions.outerbounds.plugins.apps.core._state_machine
14
13
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.app_config
14
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.capsule
15
15
 
16
16
  from .utils import TODOException as TODOException
17
17
  from .utils import safe_requests_wrapper as safe_requests_wrapper
@@ -39,24 +39,24 @@ class CapsuleStateMachine(object, metaclass=type):
39
39
  - Happy Path:
40
40
  - First time Create :
41
41
  - wait for status.updateInProgress to be set to False
42
- - (interleved) Poll the worker endpoints to check their status
42
+ - (interleaved) Poll the worker endpoints to check their status
43
43
  - showcase how many workers are coming up if things are on the cli side.
44
44
  - If the user has set some flag like `--dont-wait-to-fully-finish` then we check the `status.currentlyServedVersion` to see if even one replica is ready to
45
45
  serve traffic.
46
46
  - once the status.updateInProgress is set to False, it means that the replicas are ready
47
47
  - Upgrade:
48
48
  - wait for status.updateInProgress to be set to False
49
- - (interleved) Poll the worker endpoints to check their status and signal the user the number replicas coming up
49
+ - (interleaved) Poll the worker endpoints to check their status and signal the user the number replicas coming up
50
50
  - If the user has set some flag like `--dont-wait-to-fully-finish` then we check the `status.currentlyServedVersion` to see if even one replica is ready to
51
51
  serve traffic.
52
52
  - Unhappy Path:
53
53
  - First time Create :
54
54
  - wait for status.updateInProgress to be set to False,
55
- - (interleved) Poll the workers to check their status.
55
+ - (interleaved) Poll the workers to check their status.
56
56
  - If the worker pertaining the current deployment instance version is crashlooping then crash the deployment process with the error messages and logs.
57
57
  - Upgrade:
58
58
  - wait for status.updateInProgress to be set to False,
59
- - (interleved) Poll the workers to check their status.
59
+ - (interleaved) Poll the workers to check their status.
60
60
  - If the worker pertaining the current deployment instance version is crashlooping then crash the deployment process with the error messages and logs.
61
61
  """
62
62
  def __init__(self, capsule_id: str, current_deployment_instance_version: str):
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/click_importer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.299673 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.719100 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/code_package/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.301290 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.720371 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/code_package/code_packager.pyi CHANGED
@@ -1,15 +1,15 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.336661 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.755717 #
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.datastore.content_addressed_store
12
11
  import typing
12
+ import metaflow.datastore.content_addressed_store
13
13
 
14
14
 
15
15
  DATASTORE_SYSROOT_S3: None
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.299936 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.719350 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/cli_generator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.335049 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.754127 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/config_utils.pyi CHANGED
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.334084 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.753140 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import typing
10
+ import metaflow
11
11
  if typing.TYPE_CHECKING:
12
12
  import typing
13
13
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.config_utils
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/schema_export.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.335358 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.754436 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/typed_configs.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.293807 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.713036 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/unified_config.pyi CHANGED
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.332752 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.751692 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import typing
10
+ import metaflow
11
11
  if typing.TYPE_CHECKING:
12
12
  import typing
13
13
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.config_utils
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/dependencies.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.302059 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.721168 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/deployer.pyi CHANGED
@@ -1,19 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.264913 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.677984 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import typing
10
+ import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import datetime
13
12
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.deployer
13
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.typed_configs
14
+ import datetime
14
15
  import metaflow.mf_extensions.outerbounds.plugins.apps.core._state_machine
15
16
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.app_config
16
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.typed_configs
17
17
 
18
18
  from .config.typed_configs import TypedCoreConfig as TypedCoreConfig
19
19
  from .perimeters import PerimeterExtractor as PerimeterExtractor
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/experimental/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.301631 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.720704 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/perimeters.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.294048 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.713304 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/utils.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.301052 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.720132 #
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 logging
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.utils
13
- import metaflow.mf_extensions.outerbounds.plugins.apps.core._vendor.spinner.spinners
14
12
  import logging
13
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core._vendor.spinner.spinners
14
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.utils
15
15
 
16
16
  from ......_vendor import click as click
17
17
  from ._vendor.spinner.spinners import Spinners as Spinners
@@ -79,8 +79,8 @@ def safe_requests_wrapper(requests_module_fn, *args, conn_error_retries = 2, ret
79
79
  - How to handle retries for this case will be application specific.
80
80
  2. Errors when the API server may not be reachable (DNS resolution / network issues)
81
81
  - In this scenario, we know that something external to the API server is going wrong causing the issue.
82
- - Failing pre-maturely in the case might not be the best course of action since critical user jobs might crash on intermittent issues.
83
- - So in this case, we can just planely retry the request.
82
+ - Failing prematurely in the case might not be the best course of action since critical user jobs might crash on intermittent issues.
83
+ - So in this case, we can just plainly retry the request.
84
84
 
85
85
  This function handles the second case. It's a simple wrapper to handle the retry logic for connection errors.
86
86
  If this function is provided a `conn_error_retries` of 5, then the last retry will have waited 32 seconds.
metaflow-stubs/mf_extensions/outerbounds/plugins/aws/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.263637 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.676586 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/aws/assume_role_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.263683 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.676633 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,15 +9,16 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.user_configs.config_decorators
12
+ import metaflow.user_decorators.mutable_flow
13
+ import metaflow.user_decorators.user_flow_decorator
13
14
 
14
- from .....user_configs.config_decorators import MutableFlow as MutableFlow
15
- from .....user_configs.config_decorators import MutableStep as MutableStep
16
- from .....user_configs.config_decorators import CustomFlowDecorator as CustomFlowDecorator
15
+ from .....user_decorators.mutable_flow import MutableFlow as MutableFlow
16
+ from .....user_decorators.mutable_step import MutableStep as MutableStep
17
+ from .....user_decorators.user_flow_decorator import FlowMutator as FlowMutator
17
18
 
18
19
  OBP_ASSUME_ROLE_ARN_ENV_VAR: str
19
20
 
20
- class assume_role(metaflow.user_configs.config_decorators.CustomFlowDecorator, metaclass=type):
21
+ class assume_role(metaflow.user_decorators.user_flow_decorator.FlowMutator, metaclass=metaflow.user_decorators.user_flow_decorator.FlowMutatorMeta):
21
22
  """
22
23
  Flow-level decorator for assuming AWS IAM roles.
23
24
 
@@ -41,12 +42,15 @@ class assume_role(metaflow.user_configs.config_decorators.CustomFlowDecorator, m
41
42
  """
42
43
  def init(self, *args, **kwargs):
43
44
  ...
44
- def evaluate(self, mutable_flow: metaflow.user_configs.config_decorators.MutableFlow):
45
+ def pre_mutate(self, mutable_flow: metaflow.user_decorators.mutable_flow.MutableFlow):
45
46
  """
46
47
  This method is called by Metaflow to apply the decorator to the flow.
47
48
  It sets up environment variables that will be used by the AWS client
48
49
  to automatically assume the specified role.
49
50
  """
50
51
  ...
52
+ @classmethod
53
+ def __init_subclass__(cls_, **_kwargs):
54
+ ...
51
55
  ...
52
56
 
metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.268555 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.681720 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/async_cards.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.298353 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.717721 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import threading
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.outerbounds.plugins.card_utilities.async_cards
13
12
  import threading
13
+ import metaflow.mf_extensions.outerbounds.plugins.card_utilities.async_cards
14
14
 
15
15
  from .....metaflow_current import current as current
16
16
 
metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.268607 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.681767 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.263034 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.675906 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.263358 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.676274 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,13 +9,15 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.user_configs.config_decorators
12
+ import metaflow.user_decorators.mutable_flow
13
+ import metaflow.mf_extensions.outerbounds.plugins.checkpoint_datastores.external_chckpt
14
+ import metaflow.user_decorators.user_flow_decorator
13
15
 
14
- from .....user_configs.config_decorators import MutableFlow as MutableFlow
15
- from .....user_configs.config_decorators import MutableStep as MutableStep
16
- from .....user_configs.config_decorators import CustomFlowDecorator as CustomFlowDecorator
16
+ from .....user_decorators.user_flow_decorator import FlowMutator as FlowMutator
17
+ from .....user_decorators.mutable_flow import MutableFlow as MutableFlow
18
+ from .....user_decorators.mutable_step import MutableStep as MutableStep
17
19
 
18
- class coreweave_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecorator, metaclass=type):
20
+ class coreweave_checkpoints(metaflow.mf_extensions.outerbounds.plugins.checkpoint_datastores.external_chckpt._ExternalCheckpointFlowDeco, metaclass=metaflow.user_decorators.user_flow_decorator.FlowMutatorMeta):
19
21
  """
20
22
  This decorator is used for setting the coreweave object store as the artifact store for checkpoints/models created by the flow.
21
23
 
@@ -52,7 +54,10 @@ class coreweave_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDe
52
54
  ...
53
55
  def init(self, *args, **kwargs):
54
56
  ...
55
- def evaluate(self, mutable_flow: metaflow.user_configs.config_decorators.MutableFlow):
57
+ def pre_mutate(self, mutable_flow: metaflow.user_decorators.mutable_flow.MutableFlow):
58
+ ...
59
+ @classmethod
60
+ def __init_subclass__(cls_, **_kwargs):
56
61
  ...
57
62
  ...
58
63
 
metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.263091 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.675956 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,15 +9,15 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.user_configs.config_decorators
12
+ import metaflow.user_decorators.mutable_flow
13
+ import metaflow.mf_extensions.outerbounds.plugins.checkpoint_datastores.external_chckpt
14
+ import metaflow.user_decorators.user_flow_decorator
13
15
 
14
- from .....user_configs.config_decorators import MutableFlow as MutableFlow
15
- from .....user_configs.config_decorators import MutableStep as MutableStep
16
- from .....user_configs.config_decorators import CustomFlowDecorator as CustomFlowDecorator
16
+ from .....user_decorators.mutable_flow import MutableFlow as MutableFlow
17
17
 
18
18
  NEBIUS_ENDPOINT_URL: str
19
19
 
20
- class nebius_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecorator, metaclass=type):
20
+ class nebius_checkpoints(metaflow.mf_extensions.outerbounds.plugins.checkpoint_datastores.external_chckpt._ExternalCheckpointFlowDeco, metaclass=metaflow.user_decorators.user_flow_decorator.FlowMutatorMeta):
21
21
  """
22
22
  This decorator is used for setting the nebius's S3 compatible object store as the artifact store for
23
23
  checkpoints/models created by the flow.
@@ -58,7 +58,10 @@ class nebius_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecor
58
58
  ...
59
59
  def init(self, *args, **kwargs):
60
60
  ...
61
- def evaluate(self, mutable_flow: metaflow.user_configs.config_decorators.MutableFlow):
61
+ def pre_mutate(self, mutable_flow: metaflow.user_decorators.mutable_flow.MutableFlow):
62
+ ...
63
+ @classmethod
64
+ def __init_subclass__(cls_, **_kwargs):
62
65
  ...
63
66
  ...
64
67
 
metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.270153 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.683246 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/baker.pyi CHANGED
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.270201 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.683301 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import typing
10
+ import metaflow
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.exception
13
13
  import metaflow.mf_extensions.outerbounds.plugins.fast_bakery.fast_bakery
metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/docker_environment.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.299535 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.718949 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -54,7 +54,7 @@ class DockerEnvironment(metaflow.metaflow_environment.MetaflowEnvironment, metac
54
54
  ...
55
55
  def pylint_config(self):
56
56
  ...
57
- def get_package_commands(self, codepackage_url, datastore_type):
57
+ def get_package_commands(self, codepackage_url, datastore_type, code_package_metadata = None):
58
58
  ...
59
59
  def bootstrap_commands(self, step_name, datastore_type):
60
60
  ...
metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/fast_bakery.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.298958 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.718360 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.266640 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.679762 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/pod_killer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.266687 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.679809 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.266856 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.679961 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.266901 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.680037 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.5+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-30T20:52:28.267141 #
3
+ # MF version: 2.16.8.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-08-01T20:12:28.680288 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations