ob-metaflow-stubs 6.0.3.188rc3__py2.py3-none-any.whl → 6.0.3.188rc4__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 (243) hide show
  1. metaflow-stubs/__init__.pyi +1107 -1107
  2. metaflow-stubs/cards.pyi +1 -1
  3. metaflow-stubs/cli.pyi +1 -1
  4. metaflow-stubs/cli_components/__init__.pyi +1 -1
  5. metaflow-stubs/cli_components/utils.pyi +1 -1
  6. metaflow-stubs/client/__init__.pyi +1 -1
  7. metaflow-stubs/client/core.pyi +4 -4
  8. metaflow-stubs/client/filecache.pyi +1 -1
  9. metaflow-stubs/events.pyi +1 -1
  10. metaflow-stubs/exception.pyi +1 -1
  11. metaflow-stubs/flowspec.pyi +3 -3
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +2 -2
  14. metaflow-stubs/info_file.pyi +1 -1
  15. metaflow-stubs/metadata_provider/__init__.pyi +1 -1
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +1 -1
  17. metaflow-stubs/metadata_provider/metadata.pyi +1 -1
  18. metaflow-stubs/metadata_provider/util.pyi +1 -1
  19. metaflow-stubs/metaflow_config.pyi +1 -1
  20. metaflow-stubs/metaflow_current.pyi +40 -40
  21. metaflow-stubs/metaflow_git.pyi +1 -1
  22. metaflow-stubs/mf_extensions/__init__.pyi +1 -1
  23. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +1 -1
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +1 -1
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +1 -1
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +1 -1
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +1 -1
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +1 -1
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +1 -1
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +1 -1
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +1 -1
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +2 -2
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +1 -1
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +2 -2
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +1 -1
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +2 -2
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +3 -3
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +1 -1
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +2 -2
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +1 -1
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +1 -1
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +1 -1
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +1 -1
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +1 -1
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +1 -1
  47. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +1 -1
  48. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +1 -1
  49. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +1 -1
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +1 -1
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +1 -1
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +1 -1
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +2 -2
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +1 -1
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +2 -2
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +1 -1
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +1 -1
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +1 -1
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +1 -1
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +1 -1
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +1 -1
  62. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +1 -1
  63. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +1 -1
  64. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +1 -1
  65. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +1 -1
  66. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/__init__.pyi +1 -1
  67. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/__init__.pyi +1 -1
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_state_machine.pyi +4 -2
  69. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_vendor/__init__.pyi +1 -1
  70. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_vendor/spinner/__init__.pyi +1 -1
  71. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_vendor/spinner/spinners.pyi +1 -1
  72. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/app_cli.pyi +1 -1
  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 +1 -1
  76. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/code_package/__init__.pyi +1 -1
  77. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/code_package/code_packager.pyi +2 -2
  78. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/__init__.pyi +1 -1
  79. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/cli_generator.pyi +1 -1
  80. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/config_utils.pyi +2 -2
  81. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/schema_export.pyi +1 -1
  82. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/typed_configs.pyi +4 -2
  83. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/unified_config.pyi +1 -1
  84. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/dependencies.pyi +3 -3
  85. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/deployer.pyi +2 -2
  86. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/experimental/__init__.pyi +1 -1
  87. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/perimeters.pyi +1 -1
  88. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/utils.pyi +1 -1
  89. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/__init__.pyi +1 -1
  90. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/assume_role_decorator.pyi +1 -1
  91. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +1 -1
  92. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/async_cards.pyi +1 -1
  93. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +1 -1
  94. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +1 -1
  95. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +1 -1
  96. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +1 -1
  97. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/__init__.pyi +1 -1
  98. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/baker.pyi +2 -2
  99. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/docker_environment.pyi +2 -2
  100. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/fast_bakery.pyi +1 -1
  101. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/__init__.pyi +1 -1
  102. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/pod_killer.pyi +1 -1
  103. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +1 -1
  104. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +1 -1
  105. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +1 -1
  106. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +1 -1
  107. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/status_card.pyi +1 -1
  108. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +1 -1
  109. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +1 -1
  110. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +1 -1
  111. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +1 -1
  112. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +1 -1
  113. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +1 -1
  114. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +1 -1
  115. metaflow-stubs/multicore_utils.pyi +1 -1
  116. metaflow-stubs/ob_internal.pyi +1 -1
  117. metaflow-stubs/parameters.pyi +2 -2
  118. metaflow-stubs/plugins/__init__.pyi +12 -12
  119. metaflow-stubs/plugins/airflow/__init__.pyi +1 -1
  120. metaflow-stubs/plugins/airflow/airflow_utils.pyi +1 -1
  121. metaflow-stubs/plugins/airflow/exception.pyi +1 -1
  122. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +1 -1
  123. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +1 -1
  124. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +1 -1
  125. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +1 -1
  126. metaflow-stubs/plugins/argo/__init__.pyi +1 -1
  127. metaflow-stubs/plugins/argo/argo_client.pyi +1 -1
  128. metaflow-stubs/plugins/argo/argo_events.pyi +1 -1
  129. metaflow-stubs/plugins/argo/argo_workflows.pyi +1 -1
  130. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +1 -1
  131. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +2 -2
  132. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +1 -1
  133. metaflow-stubs/plugins/aws/__init__.pyi +1 -1
  134. metaflow-stubs/plugins/aws/aws_client.pyi +1 -1
  135. metaflow-stubs/plugins/aws/aws_utils.pyi +1 -1
  136. metaflow-stubs/plugins/aws/batch/__init__.pyi +1 -1
  137. metaflow-stubs/plugins/aws/batch/batch.pyi +1 -1
  138. metaflow-stubs/plugins/aws/batch/batch_client.pyi +1 -1
  139. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +1 -1
  140. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +1 -1
  141. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +3 -3
  142. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +1 -1
  143. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +1 -1
  144. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +1 -1
  145. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +1 -1
  146. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +1 -1
  147. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +2 -2
  148. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +1 -1
  149. metaflow-stubs/plugins/azure/__init__.pyi +1 -1
  150. metaflow-stubs/plugins/azure/azure_credential.pyi +1 -1
  151. metaflow-stubs/plugins/azure/azure_exceptions.pyi +1 -1
  152. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
  153. metaflow-stubs/plugins/azure/azure_utils.pyi +1 -1
  154. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +1 -1
  155. metaflow-stubs/plugins/azure/includefile_support.pyi +1 -1
  156. metaflow-stubs/plugins/cards/__init__.pyi +5 -5
  157. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  158. metaflow-stubs/plugins/cards/card_creator.pyi +1 -1
  159. metaflow-stubs/plugins/cards/card_datastore.pyi +1 -1
  160. metaflow-stubs/plugins/cards/card_decorator.pyi +1 -1
  161. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +1 -1
  162. metaflow-stubs/plugins/cards/card_modules/basic.pyi +1 -1
  163. metaflow-stubs/plugins/cards/card_modules/card.pyi +1 -1
  164. metaflow-stubs/plugins/cards/card_modules/components.pyi +2 -2
  165. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +1 -1
  166. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +1 -1
  167. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +1 -1
  168. metaflow-stubs/plugins/cards/card_resolver.pyi +1 -1
  169. metaflow-stubs/plugins/cards/component_serializer.pyi +1 -1
  170. metaflow-stubs/plugins/cards/exception.pyi +1 -1
  171. metaflow-stubs/plugins/catch_decorator.pyi +1 -1
  172. metaflow-stubs/plugins/datatools/__init__.pyi +1 -1
  173. metaflow-stubs/plugins/datatools/local.pyi +1 -1
  174. metaflow-stubs/plugins/datatools/s3/__init__.pyi +1 -1
  175. metaflow-stubs/plugins/datatools/s3/s3.pyi +2 -2
  176. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +1 -1
  177. metaflow-stubs/plugins/datatools/s3/s3util.pyi +1 -1
  178. metaflow-stubs/plugins/debug_logger.pyi +1 -1
  179. metaflow-stubs/plugins/debug_monitor.pyi +1 -1
  180. metaflow-stubs/plugins/environment_decorator.pyi +1 -1
  181. metaflow-stubs/plugins/events_decorator.pyi +1 -1
  182. metaflow-stubs/plugins/frameworks/__init__.pyi +1 -1
  183. metaflow-stubs/plugins/frameworks/pytorch.pyi +1 -1
  184. metaflow-stubs/plugins/gcp/__init__.pyi +1 -1
  185. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
  186. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +1 -1
  187. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +1 -1
  188. metaflow-stubs/plugins/gcp/gs_utils.pyi +1 -1
  189. metaflow-stubs/plugins/gcp/includefile_support.pyi +1 -1
  190. metaflow-stubs/plugins/kubernetes/__init__.pyi +1 -1
  191. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +1 -1
  192. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +1 -1
  193. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +1 -1
  194. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +1 -1
  195. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +1 -1
  196. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +1 -1
  197. metaflow-stubs/plugins/ollama/__init__.pyi +2 -2
  198. metaflow-stubs/plugins/parallel_decorator.pyi +1 -1
  199. metaflow-stubs/plugins/perimeters.pyi +1 -1
  200. metaflow-stubs/plugins/project_decorator.pyi +1 -1
  201. metaflow-stubs/plugins/pypi/__init__.pyi +1 -1
  202. metaflow-stubs/plugins/pypi/conda_decorator.pyi +1 -1
  203. metaflow-stubs/plugins/pypi/conda_environment.pyi +3 -3
  204. metaflow-stubs/plugins/pypi/parsers.pyi +1 -1
  205. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +1 -1
  206. metaflow-stubs/plugins/pypi/pypi_environment.pyi +1 -1
  207. metaflow-stubs/plugins/pypi/utils.pyi +1 -1
  208. metaflow-stubs/plugins/resources_decorator.pyi +1 -1
  209. metaflow-stubs/plugins/retry_decorator.pyi +1 -1
  210. metaflow-stubs/plugins/secrets/__init__.pyi +1 -1
  211. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  212. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +1 -1
  213. metaflow-stubs/plugins/snowflake/__init__.pyi +1 -1
  214. metaflow-stubs/plugins/storage_executor.pyi +1 -1
  215. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  216. metaflow-stubs/plugins/timeout_decorator.pyi +1 -1
  217. metaflow-stubs/plugins/torchtune/__init__.pyi +1 -1
  218. metaflow-stubs/plugins/uv/__init__.pyi +1 -1
  219. metaflow-stubs/plugins/uv/uv_environment.pyi +2 -2
  220. metaflow-stubs/profilers/__init__.pyi +1 -1
  221. metaflow-stubs/pylint_wrapper.pyi +1 -1
  222. metaflow-stubs/runner/__init__.pyi +1 -1
  223. metaflow-stubs/runner/deployer.pyi +4 -4
  224. metaflow-stubs/runner/deployer_impl.pyi +1 -1
  225. metaflow-stubs/runner/metaflow_runner.pyi +2 -2
  226. metaflow-stubs/runner/nbdeploy.pyi +1 -1
  227. metaflow-stubs/runner/nbrun.pyi +1 -1
  228. metaflow-stubs/runner/subprocess_manager.pyi +1 -1
  229. metaflow-stubs/runner/utils.pyi +3 -3
  230. metaflow-stubs/system/__init__.pyi +1 -1
  231. metaflow-stubs/system/system_logger.pyi +2 -2
  232. metaflow-stubs/system/system_monitor.pyi +1 -1
  233. metaflow-stubs/tagging_util.pyi +1 -1
  234. metaflow-stubs/tuple_util.pyi +1 -1
  235. metaflow-stubs/user_configs/__init__.pyi +1 -1
  236. metaflow-stubs/user_configs/config_decorators.pyi +4 -4
  237. metaflow-stubs/user_configs/config_options.pyi +2 -2
  238. metaflow-stubs/user_configs/config_parameters.pyi +5 -5
  239. {ob_metaflow_stubs-6.0.3.188rc3.dist-info → ob_metaflow_stubs-6.0.3.188rc4.dist-info}/METADATA +1 -1
  240. ob_metaflow_stubs-6.0.3.188rc4.dist-info/RECORD +243 -0
  241. ob_metaflow_stubs-6.0.3.188rc3.dist-info/RECORD +0 -243
  242. {ob_metaflow_stubs-6.0.3.188rc3.dist-info → ob_metaflow_stubs-6.0.3.188rc4.dist-info}/WHEEL +0 -0
  243. {ob_metaflow_stubs-6.0.3.188rc3.dist-info → ob_metaflow_stubs-6.0.3.188rc4.dist-info}/top_level.txt +0 -0
metaflow-stubs/cards.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.188013 #
4
+ # Generated on 2025-07-10T23:14:33.182595 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/cli.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.205328 #
4
+ # Generated on 2025-07-10T23:14:33.199927 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.204114 #
4
+ # Generated on 2025-07-10T23:14:33.198706 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.236189 #
4
+ # Generated on 2025-07-10T23:14:33.231485 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.191998 #
4
+ # Generated on 2025-07-10T23:14:33.186543 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.196659 #
4
+ # Generated on 2025-07-10T23:14:33.190964 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,11 +9,11 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.client.core
12
+ import metaflow.events
13
13
  import tarfile
14
- import datetime
15
14
  import tempfile
16
- import metaflow.events
15
+ import datetime
16
+ import metaflow.client.core
17
17
  import typing
18
18
 
19
19
  from ..metaflow_current import current as current
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.227524 #
4
+ # Generated on 2025-07-10T23:14:33.222639 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/events.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.188935 #
4
+ # Generated on 2025-07-10T23:14:33.183437 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.180092 #
4
+ # Generated on 2025-07-10T23:14:33.174481 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.184647 #
4
+ # Generated on 2025-07-10T23:14:33.179063 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,11 +9,11 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
+ import metaflow.datastore.inputs
13
+ import metaflow.unbounded_foreach
12
14
  import metaflow.exception
13
15
  import metaflow.flowspec
14
- import metaflow.unbounded_foreach
15
16
  import typing
16
- import metaflow.datastore.inputs
17
17
 
18
18
  from . import parameters as parameters
19
19
  from .parameters import DelayedEvaluationParameter as DelayedEvaluationParameter
@@ -1 +1 @@
1
- 2.15.18.1+obcheckpoint(0.2.4);ob(v1) 2025-07-10T08:45:58.174516
1
+ 2.15.18.1+obcheckpoint(0.2.4);ob(v1) 2025-07-10T23:14:33.168700
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.191352 #
4
+ # Generated on 2025-07-10T23:14:33.185871 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow._vendor.click.types
13
12
  import typing
13
+ import metaflow._vendor.click.types
14
14
  import metaflow.parameters
15
15
 
16
16
  from ._vendor import click as click
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.179417 #
4
+ # Generated on 2025-07-10T23:14:33.173760 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.183549 #
4
+ # Generated on 2025-07-10T23:14:33.177947 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.212124 #
4
+ # Generated on 2025-07-10T23:14:33.206872 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.211652 #
4
+ # Generated on 2025-07-10T23:14:33.206393 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.211866 #
4
+ # Generated on 2025-07-10T23:14:33.206608 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.181209 #
4
+ # Generated on 2025-07-10T23:14:33.175588 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,22 +1,22 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.340982 #
4
+ # Generated on 2025-07-10T23:14:33.335843 #
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
12
- import metaflow
13
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
14
11
  import metaflow.events
15
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
16
- import metaflow.metaflow_current
17
12
  import metaflow.plugins.cards.component_serializer
13
+ import metaflow
18
14
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
15
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
16
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
19
17
  import typing
18
+ import metaflow.mf_extensions.outerbounds.plugins.apps.core
19
+ import metaflow.metaflow_current
20
20
 
21
21
 
22
22
  TYPE_CHECKING: bool
@@ -229,46 +229,48 @@ class Current(object, metaclass=type):
229
229
  def graph(self):
230
230
  ...
231
231
  @property
232
- def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
232
+ def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
233
233
  """
234
- (only in the presence of the @checkpoint decorator)
234
+ (only in the presence of the @card decorator)
235
235
 
236
- The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
237
- The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
236
+ The `@card` decorator makes the cards available through the `current.card`
237
+ object. If multiple `@card` decorators are present, you can add an `ID` to
238
+ distinguish between them using `@card(id=ID)` as the decorator. You will then
239
+ be able to access that specific card using `current.card[ID].
238
240
 
239
- You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
240
- by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
241
- where the checkpoint maybe loaded or saved.
241
+ Methods available are `append` and `extend`
242
242
 
243
243
  Returns
244
- ----------
245
- CurrentCheckpointer
246
- The object for handling checkpointing within a step.
244
+ -------
245
+ CardComponentCollector
246
+ The or one of the cards attached to this step.
247
247
  """
248
248
  ...
249
249
  @property
250
- def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
250
+ def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
251
251
  """
252
- (only in the presence of the @model decorator)
252
+ (only in the presence of the @huggingface_hub decorator)
253
253
 
254
- The object used for loading / saving models.
255
- `current.model` exposes a `save` method to save models and a `load` method to load models.
256
- `current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
257
- or models loaded via `current.model.load`.
258
254
 
259
- Returns
260
- ----------
261
- ModelSerializer
262
- The object used for loading / saving models.
255
+ The `@huggingface_hub` injects a `huggingface_hub` object into the `current` object. This object provides syntactic sugar over [huggingface_hub](https://github.com/huggingface/huggingface_hub)'s [snapshot_download](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/file_download#huggingface_hub.snapshot_download) function. The `current.huggingface_hub.snapshot_download` function downloads objects from huggingface hub and saves them to the Metaflow's datastore under the `<repo_type>/<repo_id>` name. The `repo_type` is by default `model` and can be overriden by passing the `repo_type` parameter to the `snapshot_download` function.
263
256
  """
264
257
  ...
265
258
  @property
266
- def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
259
+ def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
267
260
  """
268
- (only in the presence of the @huggingface_hub decorator)
261
+ (only in the presence of the @checkpoint decorator)
262
+
263
+ The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
264
+ The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
269
265
 
266
+ You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
267
+ by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
268
+ where the checkpoint maybe loaded or saved.
270
269
 
271
- The `@huggingface_hub` injects a `huggingface_hub` object into the `current` object. This object provides syntactic sugar over [huggingface_hub](https://github.com/huggingface/huggingface_hub)'s [snapshot_download](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/file_download#huggingface_hub.snapshot_download) function. The `current.huggingface_hub.snapshot_download` function downloads objects from huggingface hub and saves them to the Metaflow's datastore under the `<repo_type>/<repo_id>` name. The `repo_type` is by default `model` and can be overriden by passing the `repo_type` parameter to the `snapshot_download` function.
270
+ Returns
271
+ ----------
272
+ CurrentCheckpointer
273
+ The object for handling checkpointing within a step.
272
274
  """
273
275
  ...
274
276
  @property
@@ -301,21 +303,19 @@ class Current(object, metaclass=type):
301
303
  """
302
304
  ...
303
305
  @property
304
- def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
306
+ def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
305
307
  """
306
- (only in the presence of the @card decorator)
307
-
308
- The `@card` decorator makes the cards available through the `current.card`
309
- object. If multiple `@card` decorators are present, you can add an `ID` to
310
- distinguish between them using `@card(id=ID)` as the decorator. You will then
311
- be able to access that specific card using `current.card[ID].
308
+ (only in the presence of the @model decorator)
312
309
 
313
- Methods available are `append` and `extend`
310
+ The object used for loading / saving models.
311
+ `current.model` exposes a `save` method to save models and a `load` method to load models.
312
+ `current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
313
+ or models loaded via `current.model.load`.
314
314
 
315
315
  Returns
316
- -------
317
- CardComponentCollector
318
- The or one of the cards attached to this step.
316
+ ----------
317
+ ModelSerializer
318
+ The object used for loading / saving models.
319
319
  """
320
320
  ...
321
321
  @property
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.188372 #
4
+ # Generated on 2025-07-10T23:14:33.182365 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.190248 #
4
+ # Generated on 2025-07-10T23:14:33.184738 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.199451 #
4
+ # Generated on 2025-07-10T23:14:33.193909 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.199512 #
4
+ # Generated on 2025-07-10T23:14:33.193971 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.199567 #
4
+ # Generated on 2025-07-10T23:14:33.194026 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.264030 #
4
+ # Generated on 2025-07-10T23:14:33.259313 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.264609 #
4
+ # Generated on 2025-07-10T23:14:33.259890 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.264078 #
4
+ # Generated on 2025-07-10T23:14:33.259360 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.311182 #
4
+ # Generated on 2025-07-10T23:14:33.306715 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.199620 #
4
+ # Generated on 2025-07-10T23:14:33.194099 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.265369 #
4
+ # Generated on 2025-07-10T23:14:33.260645 #
5
5
  ######################################################################################################
6
6
 
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.265418 #
4
+ # Generated on 2025-07-10T23:14:33.260692 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import threading
10
9
  import typing
10
+ import threading
11
11
  import metaflow
12
12
  if typing.TYPE_CHECKING:
13
13
  import threading
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.265772 #
4
+ # Generated on 2025-07-10T23:14:33.261037 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.310462 #
4
+ # Generated on 2025-07-10T23:14:33.305828 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -10,9 +10,9 @@ import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
13
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
13
14
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
14
15
  import metaflow.datastore.datastore_storage
15
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
16
16
 
17
17
  from ..exceptions import KeyNotCompatibleWithObjectException as KeyNotCompatibleWithObjectException
18
18
  from ..utils.identity_utils import pathspec_hash as pathspec_hash
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.227818 #
4
+ # Generated on 2025-07-10T23:14:33.222933 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.263169 #
4
+ # Generated on 2025-07-10T23:14:33.258458 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
12
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
12
13
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
13
14
  import metaflow
14
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
15
15
 
16
16
  from ......metaflow_current import current as current
17
17
  from ......exception import MetaflowException as MetaflowException
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.229988 #
4
+ # Generated on 2025-07-10T23:14:33.225032 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,10 +9,10 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import os
13
12
  import metaflow.decorators
14
- import metaflow
15
13
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
14
+ import os
15
+ import metaflow
16
16
 
17
17
  from .exceptions import CheckpointException as CheckpointException
18
18
  from ..utils import flowspec_utils as flowspec_utils
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.228033 #
4
+ # Generated on 2025-07-10T23:14:33.223166 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,15 +1,15 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.199661 #
4
+ # Generated on 2025-07-10T23:14:33.194147 #
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
12
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
12
+ import metaflow
13
13
 
14
14
  from ..datastructures import CheckpointArtifact as CheckpointArtifact
15
15
  from .constructors import load_checkpoint as load_checkpoint
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.266024 #
4
+ # Generated on 2025-07-10T23:14:33.261312 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-10T08:45:58.201296 #
4
+ # Generated on 2025-07-10T23:14:33.195786 #
5
5
  ######################################################################################################
6
6