ob-metaflow-stubs 6.0.4.0__py2.py3-none-any.whl → 6.0.4.1rc0__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (261) hide show
  1. metaflow-stubs/__init__.pyi +920 -914
  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 +3 -3
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +10 -10
  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 +2 -2
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +6 -2
  20. metaflow-stubs/metaflow_current.pyi +35 -35
  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 +2 -2
  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 +3 -3
  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 +5 -5
  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 +3 -3
  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 +2 -2
  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 +2 -2
  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 +4 -4
  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 +2 -2
  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 +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 +2 -2
  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 +4 -4
  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 +5 -5
  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 +3 -3
  83. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/unified_config.pyi +4 -4
  84. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/dependencies.pyi +4 -4
  85. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/deployer.pyi +3 -3
  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 +4 -4
  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 -11
  96. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +13 -11
  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 +4 -4
  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 +2 -2
  115. metaflow-stubs/multicore_utils.pyi +2 -2
  116. metaflow-stubs/ob_internal.pyi +2 -2
  117. metaflow-stubs/packaging_sys/__init__.pyi +430 -0
  118. metaflow-stubs/packaging_sys/backend.pyi +73 -0
  119. metaflow-stubs/packaging_sys/distribution_support.pyi +57 -0
  120. metaflow-stubs/packaging_sys/tar_backend.pyi +53 -0
  121. metaflow-stubs/packaging_sys/utils.pyi +26 -0
  122. metaflow-stubs/packaging_sys/v1.pyi +145 -0
  123. metaflow-stubs/parameters.pyi +4 -4
  124. metaflow-stubs/plugins/__init__.pyi +10 -10
  125. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  126. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  127. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  128. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  129. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  130. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  131. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  132. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  133. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  134. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  135. metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -3
  136. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  137. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  138. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +3 -3
  139. metaflow-stubs/plugins/argo/exit_hooks.pyi +3 -3
  140. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  141. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  142. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  143. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  144. metaflow-stubs/plugins/aws/batch/batch.pyi +4 -4
  145. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  146. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +4 -2
  147. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  148. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +3 -3
  149. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  150. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  151. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  152. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +3 -3
  153. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  154. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +3 -3
  155. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +2 -2
  156. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  157. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  158. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  159. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
  160. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  161. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  162. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  163. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  164. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  165. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  166. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  167. metaflow-stubs/plugins/cards/card_decorator.pyi +3 -5
  168. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  169. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  170. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  171. metaflow-stubs/plugins/cards/card_modules/components.pyi +3 -3
  172. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  173. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  174. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  175. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  176. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  177. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  178. metaflow-stubs/plugins/catch_decorator.pyi +2 -2
  179. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  180. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  181. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  182. metaflow-stubs/plugins/datatools/s3/s3.pyi +3 -3
  183. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  184. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  185. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  186. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  187. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  188. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  189. metaflow-stubs/plugins/exit_hook/__init__.pyi +2 -2
  190. metaflow-stubs/plugins/exit_hook/exit_hook_decorator.pyi +2 -2
  191. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  192. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  193. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  194. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
  195. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  196. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  197. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  198. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  199. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  200. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  201. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +4 -4
  202. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  203. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +4 -2
  204. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  205. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
  206. metaflow-stubs/plugins/ollama/__init__.pyi +3 -3
  207. metaflow-stubs/plugins/parallel_decorator.pyi +3 -3
  208. metaflow-stubs/plugins/perimeters.pyi +2 -2
  209. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  210. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  211. metaflow-stubs/plugins/pypi/conda_decorator.pyi +5 -8
  212. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -4
  213. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  214. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +4 -4
  215. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  216. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  217. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  218. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  219. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  220. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  221. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  222. metaflow-stubs/plugins/secrets/secrets_func.pyi +2 -2
  223. metaflow-stubs/plugins/secrets/secrets_spec.pyi +2 -2
  224. metaflow-stubs/plugins/secrets/utils.pyi +2 -2
  225. metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
  226. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  227. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +4 -4
  228. metaflow-stubs/plugins/timeout_decorator.pyi +2 -2
  229. metaflow-stubs/plugins/torchtune/__init__.pyi +2 -2
  230. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  231. metaflow-stubs/plugins/uv/uv_environment.pyi +4 -3
  232. metaflow-stubs/profilers/__init__.pyi +2 -2
  233. metaflow-stubs/pylint_wrapper.pyi +2 -2
  234. metaflow-stubs/runner/__init__.pyi +2 -2
  235. metaflow-stubs/runner/deployer.pyi +28 -28
  236. metaflow-stubs/runner/deployer_impl.pyi +3 -3
  237. metaflow-stubs/runner/metaflow_runner.pyi +5 -5
  238. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  239. metaflow-stubs/runner/nbrun.pyi +2 -2
  240. metaflow-stubs/runner/subprocess_manager.pyi +3 -2
  241. metaflow-stubs/runner/utils.pyi +3 -3
  242. metaflow-stubs/system/__init__.pyi +2 -2
  243. metaflow-stubs/system/system_logger.pyi +3 -3
  244. metaflow-stubs/system/system_monitor.pyi +2 -2
  245. metaflow-stubs/tagging_util.pyi +2 -2
  246. metaflow-stubs/tuple_util.pyi +2 -2
  247. metaflow-stubs/user_configs/__init__.pyi +2 -3
  248. metaflow-stubs/user_configs/config_options.pyi +3 -4
  249. metaflow-stubs/user_configs/config_parameters.pyi +6 -8
  250. metaflow-stubs/user_decorators/__init__.pyi +15 -0
  251. metaflow-stubs/user_decorators/common.pyi +38 -0
  252. metaflow-stubs/user_decorators/mutable_flow.pyi +223 -0
  253. metaflow-stubs/user_decorators/mutable_step.pyi +152 -0
  254. metaflow-stubs/user_decorators/user_flow_decorator.pyi +137 -0
  255. metaflow-stubs/user_decorators/user_step_decorator.pyi +323 -0
  256. {ob_metaflow_stubs-6.0.4.0.dist-info → ob_metaflow_stubs-6.0.4.1rc0.dist-info}/METADATA +1 -1
  257. ob_metaflow_stubs-6.0.4.1rc0.dist-info/RECORD +260 -0
  258. metaflow-stubs/user_configs/config_decorators.pyi +0 -251
  259. ob_metaflow_stubs-6.0.4.0.dist-info/RECORD +0 -249
  260. {ob_metaflow_stubs-6.0.4.0.dist-info → ob_metaflow_stubs-6.0.4.1rc0.dist-info}/WHEEL +0 -0
  261. {ob_metaflow_stubs-6.0.4.0.dist-info → ob_metaflow_stubs-6.0.4.1rc0.dist-info}/top_level.txt +0 -0
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.599805 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.663927 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/config_utils.pyi CHANGED
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.598830 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.662941 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.config_utils
14
12
  import metaflow._vendor.click.types
13
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.config_utils
14
+ import typing
15
15
 
16
16
  from ......._vendor import click as click
17
17
 
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.600127 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.664237 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.558435 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.622654 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import typing
10
10
  from typing import TypedDict
11
11
  if typing.TYPE_CHECKING:
12
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.typed_configs
12
13
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.unified_config
13
14
  import typing
14
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.typed_configs
15
15
 
16
16
  from .unified_config import CoreConfig as CoreConfig
17
17
 
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/unified_config.pyi CHANGED
@@ -1,16 +1,16 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.597494 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.661486 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
12
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.config_utils
13
+ import typing
14
14
 
15
15
  from .config_utils import ConfigField as ConfigField
16
16
  from .config_utils import ConfigMeta as ConfigMeta
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/dependencies.pyi CHANGED
@@ -1,16 +1,16 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.566435 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.630674 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import typing
12
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.app_config
13
11
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.dependencies
12
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.app_config
13
+ import typing
14
14
 
15
15
  from .app_config import AppConfig as AppConfig
16
16
  from .utils import TODOException as TODOException
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/deployer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.529328 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.587815 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.app_config
13
12
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.config.typed_configs
13
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.app_config
14
14
  import metaflow.mf_extensions.outerbounds.plugins.apps.core.deployer
15
15
 
16
16
  from .config.typed_configs import TypedCoreConfig as TypedCoreConfig
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.565978 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.630239 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.564777 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.628885 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/utils.pyi CHANGED
@@ -1,16 +1,16 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.565418 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.629652 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import logging
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.outerbounds.plugins.apps.core.utils
13
12
  import metaflow.mf_extensions.outerbounds.plugins.apps.core._vendor.spinner.spinners
13
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core.utils
14
14
  import logging
15
15
 
16
16
  from ......_vendor import click as click
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.528486 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.586966 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.528537 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.587013 #
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.user_flow_decorator
13
+ import metaflow.user_decorators.mutable_flow
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.user_flow_decorator import FlowMutator as FlowMutator
16
+ from .....user_decorators.mutable_flow import MutableFlow as MutableFlow
17
+ from .....user_decorators.mutable_step import MutableStep as MutableStep
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.533005 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.591556 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.562898 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.626975 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.533052 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.591603 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.527872 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.586292 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.528205 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.586644 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,13 +9,14 @@ 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.user_flow_decorator
13
+ import metaflow.user_decorators.mutable_flow
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.user_flow_decorator import FlowMutator as FlowMutator
16
+ from .....user_decorators.mutable_flow import MutableFlow as MutableFlow
17
+ from .....user_decorators.mutable_step import MutableStep as MutableStep
17
18
 
18
- class coreweave_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecorator, metaclass=type):
19
+ class coreweave_checkpoints(metaflow.user_decorators.user_flow_decorator.FlowMutator, metaclass=metaflow.user_decorators.user_flow_decorator.FlowMutatorMeta):
19
20
  """
20
21
  This decorator is used for setting the coreweave object store as the artifact store for checkpoints/models created by the flow.
21
22
 
@@ -35,7 +36,7 @@ class coreweave_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDe
35
36
  ```python
36
37
  from metaflow import checkpoint, step, FlowSpec, coreweave_checkpoints
37
38
 
38
- @coreweave_checkpoints(secrets=[], bucket_path=None)
39
+ @coreweave_checkpoints(secrets=[], bucket_path="s3://my-coreweave-bucket/foo")
39
40
  class MyFlow(FlowSpec):
40
41
  @checkpoint
41
42
  @step
@@ -48,11 +49,12 @@ class coreweave_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDe
48
49
  pass
49
50
  ```
50
51
  """
51
- def __init__(self, *args, **kwargs):
52
- ...
53
52
  def init(self, *args, **kwargs):
54
53
  ...
55
- def evaluate(self, mutable_flow: metaflow.user_configs.config_decorators.MutableFlow):
54
+ def pre_mutate(self, mutable_flow: metaflow.user_decorators.mutable_flow.MutableFlow):
55
+ ...
56
+ @classmethod
57
+ def __init_subclass__(cls_, **_kwargs):
56
58
  ...
57
59
  ...
58
60
 
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.527918 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.586357 #
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.user_flow_decorator
13
+ import metaflow.user_decorators.mutable_flow
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.user_flow_decorator import FlowMutator as FlowMutator
16
+ from .....user_decorators.mutable_flow import MutableFlow as MutableFlow
17
+ from .....user_decorators.mutable_step import MutableStep as MutableStep
17
18
 
18
19
  NEBIUS_ENDPOINT_URL: str
19
20
 
20
- class nebius_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecorator, metaclass=type):
21
+ class nebius_checkpoints(metaflow.user_decorators.user_flow_decorator.FlowMutator, metaclass=metaflow.user_decorators.user_flow_decorator.FlowMutatorMeta):
21
22
  """
22
23
  This decorator is used for setting the nebius's S3 compatible object store as the artifact store for
23
24
  checkpoints/models created by the flow.
@@ -41,7 +42,7 @@ class nebius_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecor
41
42
  ```python
42
43
  from metaflow import checkpoint, step, FlowSpec, nebius_checkpoints
43
44
 
44
- @nebius_checkpoints(secrets=[], bucket_path=None)
45
+ @nebius_checkpoints(secrets=[], bucket_path="s3://my-nebius-bucket/foo")
45
46
  class MyFlow(FlowSpec):
46
47
  @checkpoint
47
48
  @step
@@ -54,11 +55,12 @@ class nebius_checkpoints(metaflow.user_configs.config_decorators.CustomFlowDecor
54
55
  pass
55
56
  ```
56
57
  """
57
- def __init__(self, *args, **kwargs):
58
- ...
59
58
  def init(self, *args, **kwargs):
60
59
  ...
61
- def evaluate(self, mutable_flow: metaflow.user_configs.config_decorators.MutableFlow):
60
+ def pre_mutate(self, mutable_flow: metaflow.user_decorators.mutable_flow.MutableFlow):
61
+ ...
62
+ @classmethod
63
+ def __init_subclass__(cls_, **_kwargs):
62
64
  ...
63
65
  ...
64
66
 
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.534489 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.593036 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/baker.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.534535 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.593082 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
12
  import metaflow.mf_extensions.outerbounds.plugins.fast_bakery.fast_bakery
13
+ import typing
14
14
  import metaflow.exception
15
15
 
16
16
  from .....exception import MetaflowException as MetaflowException
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.564092 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.628191 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.exception
13
12
  import metaflow.metaflow_environment
13
+ import metaflow.exception
14
14
 
15
15
  from .....exception import MetaflowException as MetaflowException
16
16
  from .....metaflow_config import get_pinned_conda_libs as get_pinned_conda_libs
@@ -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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.563520 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.627609 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.531050 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.589598 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.531096 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.589646 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.531269 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.589796 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.531314 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.589844 #
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.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.531563 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.590088 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.532325 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.590855 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/status_card.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.532772 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.591298 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.527511 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.585921 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.527554 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.585965 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi CHANGED
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.534003 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.592555 #
5
5
  ######################################################################################################
6
6
 
metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.534048 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.592602 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-11T23:29:18.557766 #
3
+ # MF version: 2.16.0.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-14T20:03:25.621980 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations