ob-metaflow-stubs 6.0.3.188rc0__py2.py3-none-any.whl → 6.0.3.188rc2__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 (244) hide show
  1. metaflow-stubs/__init__.pyi +904 -904
  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 +2 -2
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +1 -1
  11. metaflow-stubs/flowspec.pyi +2 -2
  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 +38 -38
  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 +2 -2
  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 +3 -3
  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 +3 -3
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +1 -1
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +3 -3
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +3 -3
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +1 -1
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +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 +2 -2
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +1 -1
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +1 -1
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +2 -2
  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 +2 -2
  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 +2 -2
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +1 -1
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +1 -1
  62. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +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 +4 -6
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/_state_machine.pyi +1 -1
  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 +5 -45
  73. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/app_config.pyi +15 -30
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/capsule.pyi +3 -3
  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 +4 -2
  78. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/__init__.pyi +22 -0
  79. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/cli_generator.pyi +53 -0
  80. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/config_utils.pyi +407 -0
  81. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/schema_export.pyi +64 -0
  82. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/config/unified_config.pyi +214 -0
  83. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/dependencies.pyi +3 -3
  84. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/experimental/__init__.pyi +1 -11
  85. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/perimeters.pyi +1 -1
  86. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/utils.pyi +2 -2
  87. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/__init__.pyi +1 -1
  88. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/assume_role_decorator.pyi +1 -1
  89. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +1 -1
  90. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/async_cards.pyi +2 -2
  91. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +1 -1
  92. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +1 -1
  93. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +1 -1
  94. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +1 -1
  95. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/__init__.pyi +1 -1
  96. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/baker.pyi +2 -2
  97. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/docker_environment.pyi +1 -1
  98. metaflow-stubs/mf_extensions/outerbounds/plugins/fast_bakery/fast_bakery.pyi +1 -1
  99. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/__init__.pyi +1 -1
  100. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/pod_killer.pyi +1 -1
  101. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +1 -1
  102. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +1 -1
  103. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +1 -1
  104. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +1 -1
  105. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/status_card.pyi +1 -1
  106. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +1 -1
  107. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +1 -1
  108. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +1 -1
  109. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +1 -1
  110. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +1 -1
  111. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +1 -1
  112. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +1 -1
  113. metaflow-stubs/multicore_utils.pyi +1 -1
  114. metaflow-stubs/ob_internal.pyi +1 -1
  115. metaflow-stubs/parameters.pyi +2 -2
  116. metaflow-stubs/plugins/__init__.pyi +9 -9
  117. metaflow-stubs/plugins/airflow/__init__.pyi +1 -1
  118. metaflow-stubs/plugins/airflow/airflow_utils.pyi +1 -1
  119. metaflow-stubs/plugins/airflow/exception.pyi +1 -1
  120. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +1 -1
  121. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +1 -1
  122. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +1 -1
  123. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +1 -1
  124. metaflow-stubs/plugins/argo/__init__.pyi +1 -1
  125. metaflow-stubs/plugins/argo/argo_client.pyi +1 -1
  126. metaflow-stubs/plugins/argo/argo_events.pyi +1 -1
  127. metaflow-stubs/plugins/argo/argo_workflows.pyi +2 -2
  128. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +2 -2
  129. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +2 -2
  130. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +2 -2
  131. metaflow-stubs/plugins/aws/__init__.pyi +1 -1
  132. metaflow-stubs/plugins/aws/aws_client.pyi +1 -1
  133. metaflow-stubs/plugins/aws/aws_utils.pyi +1 -1
  134. metaflow-stubs/plugins/aws/batch/__init__.pyi +1 -1
  135. metaflow-stubs/plugins/aws/batch/batch.pyi +1 -1
  136. metaflow-stubs/plugins/aws/batch/batch_client.pyi +1 -1
  137. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +1 -1
  138. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +1 -1
  139. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +3 -3
  140. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +1 -1
  141. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +1 -1
  142. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +1 -1
  143. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +1 -1
  144. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +1 -1
  145. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +2 -2
  146. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +2 -2
  147. metaflow-stubs/plugins/azure/__init__.pyi +1 -1
  148. metaflow-stubs/plugins/azure/azure_credential.pyi +1 -1
  149. metaflow-stubs/plugins/azure/azure_exceptions.pyi +1 -1
  150. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
  151. metaflow-stubs/plugins/azure/azure_utils.pyi +1 -1
  152. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +1 -1
  153. metaflow-stubs/plugins/azure/includefile_support.pyi +1 -1
  154. metaflow-stubs/plugins/cards/__init__.pyi +1 -1
  155. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  156. metaflow-stubs/plugins/cards/card_creator.pyi +1 -1
  157. metaflow-stubs/plugins/cards/card_datastore.pyi +1 -1
  158. metaflow-stubs/plugins/cards/card_decorator.pyi +1 -1
  159. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +1 -1
  160. metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
  161. metaflow-stubs/plugins/cards/card_modules/card.pyi +1 -1
  162. metaflow-stubs/plugins/cards/card_modules/components.pyi +2 -2
  163. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +1 -1
  164. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +1 -1
  165. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +1 -1
  166. metaflow-stubs/plugins/cards/card_resolver.pyi +1 -1
  167. metaflow-stubs/plugins/cards/component_serializer.pyi +1 -1
  168. metaflow-stubs/plugins/cards/exception.pyi +1 -1
  169. metaflow-stubs/plugins/catch_decorator.pyi +2 -2
  170. metaflow-stubs/plugins/datatools/__init__.pyi +1 -1
  171. metaflow-stubs/plugins/datatools/local.pyi +1 -1
  172. metaflow-stubs/plugins/datatools/s3/__init__.pyi +1 -1
  173. metaflow-stubs/plugins/datatools/s3/s3.pyi +1 -1
  174. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +1 -1
  175. metaflow-stubs/plugins/datatools/s3/s3util.pyi +1 -1
  176. metaflow-stubs/plugins/debug_logger.pyi +1 -1
  177. metaflow-stubs/plugins/debug_monitor.pyi +1 -1
  178. metaflow-stubs/plugins/environment_decorator.pyi +1 -1
  179. metaflow-stubs/plugins/events_decorator.pyi +1 -1
  180. metaflow-stubs/plugins/frameworks/__init__.pyi +1 -1
  181. metaflow-stubs/plugins/frameworks/pytorch.pyi +1 -1
  182. metaflow-stubs/plugins/gcp/__init__.pyi +1 -1
  183. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
  184. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +1 -1
  185. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +1 -1
  186. metaflow-stubs/plugins/gcp/gs_utils.pyi +1 -1
  187. metaflow-stubs/plugins/gcp/includefile_support.pyi +1 -1
  188. metaflow-stubs/plugins/kubernetes/__init__.pyi +1 -1
  189. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +1 -1
  190. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +1 -1
  191. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +1 -1
  192. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +1 -1
  193. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +1 -1
  194. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +1 -1
  195. metaflow-stubs/plugins/ollama/__init__.pyi +2 -2
  196. metaflow-stubs/plugins/parallel_decorator.pyi +1 -1
  197. metaflow-stubs/plugins/perimeters.pyi +1 -1
  198. metaflow-stubs/plugins/project_decorator.pyi +1 -1
  199. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  200. metaflow-stubs/plugins/pypi/conda_decorator.pyi +1 -1
  201. metaflow-stubs/plugins/pypi/conda_environment.pyi +2 -2
  202. metaflow-stubs/plugins/pypi/parsers.pyi +1 -1
  203. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +1 -1
  204. metaflow-stubs/plugins/pypi/pypi_environment.pyi +1 -1
  205. metaflow-stubs/plugins/pypi/utils.pyi +1 -1
  206. metaflow-stubs/plugins/resources_decorator.pyi +1 -1
  207. metaflow-stubs/plugins/retry_decorator.pyi +1 -1
  208. metaflow-stubs/plugins/secrets/__init__.pyi +1 -1
  209. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  210. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +1 -1
  211. metaflow-stubs/plugins/snowflake/__init__.pyi +1 -1
  212. metaflow-stubs/plugins/storage_executor.pyi +1 -1
  213. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  214. metaflow-stubs/plugins/timeout_decorator.pyi +2 -2
  215. metaflow-stubs/plugins/torchtune/__init__.pyi +1 -1
  216. metaflow-stubs/plugins/uv/__init__.pyi +1 -1
  217. metaflow-stubs/plugins/uv/uv_environment.pyi +1 -1
  218. metaflow-stubs/profilers/__init__.pyi +1 -1
  219. metaflow-stubs/pylint_wrapper.pyi +1 -1
  220. metaflow-stubs/runner/__init__.pyi +1 -1
  221. metaflow-stubs/runner/deployer.pyi +28 -28
  222. metaflow-stubs/runner/deployer_impl.pyi +1 -1
  223. metaflow-stubs/runner/metaflow_runner.pyi +1 -1
  224. metaflow-stubs/runner/nbdeploy.pyi +1 -1
  225. metaflow-stubs/runner/nbrun.pyi +1 -1
  226. metaflow-stubs/runner/subprocess_manager.pyi +1 -1
  227. metaflow-stubs/runner/utils.pyi +2 -2
  228. metaflow-stubs/system/__init__.pyi +1 -1
  229. metaflow-stubs/system/system_logger.pyi +1 -1
  230. metaflow-stubs/system/system_monitor.pyi +1 -1
  231. metaflow-stubs/tagging_util.pyi +1 -1
  232. metaflow-stubs/tuple_util.pyi +1 -1
  233. metaflow-stubs/user_configs/__init__.pyi +1 -1
  234. metaflow-stubs/user_configs/config_decorators.pyi +4 -4
  235. metaflow-stubs/user_configs/config_options.pyi +1 -1
  236. metaflow-stubs/user_configs/config_parameters.pyi +5 -5
  237. {ob_metaflow_stubs-6.0.3.188rc0.dist-info → ob_metaflow_stubs-6.0.3.188rc2.dist-info}/METADATA +1 -1
  238. ob_metaflow_stubs-6.0.3.188rc2.dist-info/RECORD +241 -0
  239. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/cli_to_config.pyi +0 -17
  240. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/secrets.pyi +0 -46
  241. metaflow-stubs/mf_extensions/outerbounds/plugins/apps/core/validations.pyi +0 -24
  242. ob_metaflow_stubs-6.0.3.188rc0.dist-info/RECORD +0 -239
  243. {ob_metaflow_stubs-6.0.3.188rc0.dist-info → ob_metaflow_stubs-6.0.3.188rc2.dist-info}/WHEEL +0 -0
  244. {ob_metaflow_stubs-6.0.3.188rc0.dist-info → ob_metaflow_stubs-6.0.3.188rc2.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-03T01:34:48.285346 #
4
+ # Generated on 2025-07-09T05:17:09.502954 #
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-03T01:34:48.301536 #
4
+ # Generated on 2025-07-09T05:17:09.519274 #
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-03T01:34:48.300386 #
4
+ # Generated on 2025-07-09T05:17:09.518128 #
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-03T01:34:48.328512 #
4
+ # Generated on 2025-07-09T05:17:09.546073 #
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-03T01:34:48.289176 #
4
+ # Generated on 2025-07-09T05:17:09.506860 #
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-03T01:34:48.293581 #
4
+ # Generated on 2025-07-09T05:17:09.511352 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,12 +9,12 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import datetime
13
12
  import metaflow.events
14
- import typing
15
- import tarfile
13
+ import datetime
16
14
  import metaflow.client.core
17
15
  import tempfile
16
+ import tarfile
17
+ import typing
18
18
 
19
19
  from ..metaflow_current import current as current
20
20
  from ..events import Trigger as Trigger
@@ -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-03T01:34:48.323620 #
4
+ # Generated on 2025-07-09T05:17:09.541235 #
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.datastore.content_addressed_store
13
+ import metaflow.exception
14
14
 
15
15
  from ..exception import MetaflowException as MetaflowException
16
16
 
metaflow-stubs/events.pyi CHANGED
@@ -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-03T01:34:48.286090 #
4
+ # Generated on 2025-07-09T05:17:09.503639 #
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.events
12
11
  import metaflow
12
+ import metaflow.events
13
13
 
14
14
 
15
15
  TYPE_CHECKING: bool
@@ -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-03T01:34:48.277267 #
4
+ # Generated on 2025-07-09T05:17:09.494914 #
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-03T01:34:48.281871 #
4
+ # Generated on 2025-07-09T05:17:09.499523 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.exception
13
12
  import typing
13
+ import metaflow.exception
14
14
  import metaflow.unbounded_foreach
15
15
  import metaflow.datastore.inputs
16
16
  import metaflow.flowspec
@@ -1 +1 @@
1
- 2.15.18.1+obcheckpoint(0.2.4);ob(v1) 2025-07-03T01:34:48.271686
1
+ 2.15.18.1+obcheckpoint(0.2.4);ob(v1) 2025-07-09T05:17:09.489421
@@ -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-03T01:34:48.288507 #
4
+ # Generated on 2025-07-09T05:17:09.506227 #
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.parameters
13
12
  import typing
13
+ import metaflow.parameters
14
14
  import metaflow._vendor.click.types
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-03T01:34:48.276547 #
4
+ # Generated on 2025-07-09T05:17:09.494240 #
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-03T01:34:48.280802 #
4
+ # Generated on 2025-07-09T05:17:09.498340 #
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-03T01:34:48.308426 #
4
+ # Generated on 2025-07-09T05:17:09.526012 #
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-03T01:34:48.307954 #
4
+ # Generated on 2025-07-09T05:17:09.525548 #
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-03T01:34:48.308172 #
4
+ # Generated on 2025-07-09T05:17:09.525758 #
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-03T01:34:48.278377 #
4
+ # Generated on 2025-07-09T05:17:09.496038 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,21 +1,21 @@
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-03T01:34:48.433033 #
4
+ # Generated on 2025-07-09T05:17:09.661559 #
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.plugins.cards.component_serializer
11
12
  import metaflow.events
12
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
13
- import typing
14
13
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
14
+ import typing
15
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
15
16
  import metaflow
16
- import metaflow.plugins.cards.component_serializer
17
- import metaflow.metaflow_current
18
17
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
18
+ import metaflow.metaflow_current
19
19
 
20
20
 
21
21
  TYPE_CHECKING: bool
@@ -228,32 +228,12 @@ class Current(object, metaclass=type):
228
228
  def graph(self):
229
229
  ...
230
230
  @property
231
- def parallel(self) -> "metaflow.metaflow_current.Parallel":
231
+ def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
232
232
  """
233
- (only in the presence of the @parallel decorator)
234
-
235
- Returns a namedtuple with relevant information about the parallel task.
233
+ (only in the presence of the @huggingface_hub decorator)
236
234
 
237
- Returns
238
- -------
239
- Parallel
240
- `namedtuple` with the following fields:
241
- - main_ip (`str`)
242
- The IP address of the control task.
243
- - num_nodes (`int`)
244
- The total number of tasks created by @parallel
245
- - node_index (`int`)
246
- The index of the current task in all the @parallel tasks.
247
- - control_task_id (`Optional[str]`)
248
- The task ID of the control task. Available to all tasks.
249
- """
250
- ...
251
- @property
252
- def is_parallel(self) -> bool:
253
- """
254
- (only in the presence of the @parallel decorator)
255
235
 
256
- True if the current step is a @parallel step.
236
+ 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.
257
237
  """
258
238
  ...
259
239
  @property
@@ -273,15 +253,6 @@ class Current(object, metaclass=type):
273
253
  """
274
254
  ...
275
255
  @property
276
- def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
277
- """
278
- (only in the presence of the @huggingface_hub decorator)
279
-
280
-
281
- 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.
282
- """
283
- ...
284
- @property
285
256
  def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
286
257
  """
287
258
  (only in the presence of the @checkpoint decorator)
@@ -300,6 +271,35 @@ class Current(object, metaclass=type):
300
271
  """
301
272
  ...
302
273
  @property
274
+ def parallel(self) -> "metaflow.metaflow_current.Parallel":
275
+ """
276
+ (only in the presence of the @parallel decorator)
277
+
278
+ Returns a namedtuple with relevant information about the parallel task.
279
+
280
+ Returns
281
+ -------
282
+ Parallel
283
+ `namedtuple` with the following fields:
284
+ - main_ip (`str`)
285
+ The IP address of the control task.
286
+ - num_nodes (`int`)
287
+ The total number of tasks created by @parallel
288
+ - node_index (`int`)
289
+ The index of the current task in all the @parallel tasks.
290
+ - control_task_id (`Optional[str]`)
291
+ The task ID of the control task. Available to all tasks.
292
+ """
293
+ ...
294
+ @property
295
+ def is_parallel(self) -> bool:
296
+ """
297
+ (only in the presence of the @parallel decorator)
298
+
299
+ True if the current step is a @parallel step.
300
+ """
301
+ ...
302
+ @property
303
303
  def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
304
304
  """
305
305
  (only in the presence of the @card decorator)
@@ -320,7 +320,7 @@ class Current(object, metaclass=type):
320
320
  @property
321
321
  def trigger(self) -> "metaflow.events.Trigger":
322
322
  """
323
- (only in the presence of the @trigger_on_finish, or @trigger decorators)
323
+ (only in the presence of the @trigger, or @trigger_on_finish decorators)
324
324
 
325
325
  Returns `Trigger` if the current run is triggered by an event
326
326
 
@@ -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-03T01:34:48.285104 #
4
+ # Generated on 2025-07-09T05:17:09.502729 #
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-03T01:34:48.287375 #
4
+ # Generated on 2025-07-09T05:17:09.505150 #
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-03T01:34:48.296352 #
4
+ # Generated on 2025-07-09T05:17:09.514146 #
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-03T01:34:48.296417 #
4
+ # Generated on 2025-07-09T05:17:09.514235 #
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-03T01:34:48.296472 #
4
+ # Generated on 2025-07-09T05:17:09.514305 #
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-03T01:34:48.358502 #
4
+ # Generated on 2025-07-09T05:17:09.575608 #
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-03T01:34:48.359187 #
4
+ # Generated on 2025-07-09T05:17:09.576163 #
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 threading
13
12
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
13
+ import threading
14
14
 
15
15
  from ......metaflow_current import current as current
16
16
 
@@ -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-03T01:34:48.358549 #
4
+ # Generated on 2025-07-09T05:17:09.575654 #
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-03T01:34:48.403963 #
4
+ # Generated on 2025-07-09T05:17:09.632315 #
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-03T01:34:48.296525 #
4
+ # Generated on 2025-07-09T05:17:09.514361 #
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-03T01:34:48.359927 #
4
+ # Generated on 2025-07-09T05:17:09.576911 #
5
5
  ######################################################################################################
6
6
 
@@ -1,18 +1,18 @@
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-03T01:34:48.360013 #
4
+ # Generated on 2025-07-09T05:17:09.576958 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import threading
10
9
  import typing
11
10
  import metaflow
11
+ import threading
12
12
  if typing.TYPE_CHECKING:
13
- import threading
14
13
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
15
14
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
15
+ import threading
16
16
 
17
17
  from ...card_utils.deco_injection_mixin import CardDecoratorInjector as CardDecoratorInjector
18
18
  from ...card_utils.async_cards import CardRefresher as CardRefresher
@@ -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-03T01:34:48.360361 #
4
+ # Generated on 2025-07-09T05:17:09.577332 #
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-03T01:34:48.403235 #
4
+ # Generated on 2025-07-09T05:17:09.631576 #
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 metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
13
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
14
12
  import metaflow.datastore.datastore_storage
13
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
15
14
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_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-03T01:34:48.323914 #
4
+ # Generated on 2025-07-09T05:17:09.541525 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,17 +1,17 @@
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-03T01:34:48.357635 #
4
+ # Generated on 2025-07-09T05:17:09.574753 #
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.obcheckpoint.plugins.machine_learning_utilities.datastructures
12
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
13
11
  import metaflow
14
12
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
13
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
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-03T01:34:48.326010 #
4
+ # Generated on 2025-07-09T05:17:09.543579 #
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
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
14
12
  import metaflow.decorators
15
13
  import metaflow
14
+ import os
15
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
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-03T01:34:48.324152 #
4
+ # Generated on 2025-07-09T05:17:09.541736 #
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-03T01:34:48.296567 #
4
+ # Generated on 2025-07-09T05:17:09.514406 #
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.obcheckpoint.plugins.machine_learning_utilities.datastructures
12
11
  import metaflow
12
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
13
13
 
14
14
  from ..datastructures import CheckpointArtifact as CheckpointArtifact
15
15
  from .constructors import load_checkpoint as load_checkpoint