ob-metaflow-stubs 6.0.3.186__py2.py3-none-any.whl → 6.0.3.188__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 (222) hide show
  1. metaflow-stubs/__init__.pyi +878 -862
  2. metaflow-stubs/cards.pyi +2 -2
  3. metaflow-stubs/cli.pyi +2 -2
  4. metaflow-stubs/cli_components/__init__.pyi +2 -2
  5. metaflow-stubs/cli_components/utils.pyi +2 -2
  6. metaflow-stubs/client/__init__.pyi +2 -2
  7. metaflow-stubs/client/core.pyi +6 -6
  8. metaflow-stubs/client/filecache.pyi +3 -3
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +5 -5
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +4 -4
  14. metaflow-stubs/info_file.pyi +2 -2
  15. metaflow-stubs/metadata_provider/__init__.pyi +2 -2
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +2 -2
  17. metaflow-stubs/metadata_provider/metadata.pyi +3 -3
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +2 -2
  20. metaflow-stubs/metaflow_current.pyi +39 -39
  21. metaflow-stubs/metaflow_git.pyi +2 -2
  22. metaflow-stubs/mf_extensions/__init__.pyi +2 -2
  23. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +2 -2
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +2 -2
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +2 -2
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +2 -2
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +2 -2
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +2 -2
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +3 -3
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +2 -2
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +2 -2
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +4 -4
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +2 -2
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +5 -5
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +2 -2
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +4 -4
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +4 -4
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +3 -3
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +2 -2
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +3 -3
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +3 -3
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +2 -2
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +3 -3
  47. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +2 -2
  48. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +3 -3
  49. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +2 -2
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +2 -2
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +3 -3
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +2 -2
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +3 -3
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +2 -2
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +5 -5
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +2 -2
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +2 -2
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +2 -2
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +2 -2
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +2 -2
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +2 -2
  62. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +3 -3
  63. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +3 -3
  64. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +2 -2
  65. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +2 -2
  66. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/__init__.pyi +2 -2
  67. metaflow-stubs/mf_extensions/outerbounds/plugins/aws/assume_role_decorator.pyi +2 -2
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +2 -2
  69. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/async_cards.pyi +2 -2
  70. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +2 -2
  71. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +2 -2
  72. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +2 -2
  73. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +2 -2
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/__init__.pyi +2 -2
  75. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/pod_killer.pyi +2 -2
  76. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +2 -2
  77. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +2 -2
  78. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +2 -2
  79. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +2 -2
  80. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/status_card.pyi +2 -2
  81. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +2 -2
  82. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +2 -2
  83. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
  84. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
  85. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +2 -2
  86. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
  87. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +2 -2
  88. metaflow-stubs/multicore_utils.pyi +2 -2
  89. metaflow-stubs/ob_internal.pyi +2 -2
  90. metaflow-stubs/parameters.pyi +4 -4
  91. metaflow-stubs/plugins/__init__.pyi +14 -12
  92. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  93. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  94. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  95. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  96. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  97. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  98. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  99. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  100. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  101. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  102. metaflow-stubs/plugins/argo/argo_workflows.pyi +6 -33
  103. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  104. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  105. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +4 -4
  106. metaflow-stubs/plugins/argo/exit_hooks.pyi +45 -0
  107. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  108. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  109. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  110. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  111. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  112. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  113. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
  114. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  115. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +4 -4
  116. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  117. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  118. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  119. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  120. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  121. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +4 -4
  122. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
  123. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  124. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  125. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  126. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +4 -4
  127. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  128. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  129. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  130. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  131. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  132. metaflow-stubs/plugins/cards/card_creator.pyi +4 -3
  133. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  134. metaflow-stubs/plugins/cards/card_decorator.pyi +13 -2
  135. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  136. metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
  137. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  138. metaflow-stubs/plugins/cards/card_modules/components.pyi +5 -5
  139. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  140. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  141. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +10 -2
  142. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  143. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  144. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  145. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  146. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  147. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  148. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  149. metaflow-stubs/plugins/datatools/s3/s3.pyi +3 -3
  150. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  151. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  152. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  153. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  154. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  155. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  156. metaflow-stubs/plugins/exit_hook/__init__.pyi +11 -0
  157. metaflow-stubs/plugins/exit_hook/exit_hook_decorator.pyi +20 -0
  158. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  159. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  160. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  161. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +4 -4
  162. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  163. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  164. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  165. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  166. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  167. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +3 -3
  168. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
  169. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  170. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
  171. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  172. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
  173. metaflow-stubs/plugins/ollama/__init__.pyi +3 -3
  174. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  175. metaflow-stubs/plugins/perimeters.pyi +2 -2
  176. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  177. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  178. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  179. metaflow-stubs/plugins/pypi/conda_environment.pyi +4 -4
  180. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  181. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  182. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  183. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  184. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  185. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  186. metaflow-stubs/plugins/secrets/__init__.pyi +7 -3
  187. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  188. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +9 -49
  189. metaflow-stubs/plugins/secrets/secrets_func.pyi +31 -0
  190. metaflow-stubs/plugins/secrets/secrets_spec.pyi +42 -0
  191. metaflow-stubs/plugins/secrets/utils.pyi +28 -0
  192. metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
  193. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  194. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
  195. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  196. metaflow-stubs/plugins/torchtune/__init__.pyi +2 -2
  197. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  198. metaflow-stubs/plugins/uv/uv_environment.pyi +3 -3
  199. metaflow-stubs/profilers/__init__.pyi +2 -2
  200. metaflow-stubs/pylint_wrapper.pyi +2 -2
  201. metaflow-stubs/runner/__init__.pyi +2 -2
  202. metaflow-stubs/runner/deployer.pyi +29 -29
  203. metaflow-stubs/runner/deployer_impl.pyi +2 -2
  204. metaflow-stubs/runner/metaflow_runner.pyi +4 -4
  205. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  206. metaflow-stubs/runner/nbrun.pyi +2 -2
  207. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  208. metaflow-stubs/runner/utils.pyi +3 -3
  209. metaflow-stubs/system/__init__.pyi +2 -2
  210. metaflow-stubs/system/system_logger.pyi +2 -2
  211. metaflow-stubs/system/system_monitor.pyi +2 -2
  212. metaflow-stubs/tagging_util.pyi +2 -2
  213. metaflow-stubs/tuple_util.pyi +2 -2
  214. metaflow-stubs/user_configs/__init__.pyi +2 -2
  215. metaflow-stubs/user_configs/config_decorators.pyi +6 -6
  216. metaflow-stubs/user_configs/config_options.pyi +3 -3
  217. metaflow-stubs/user_configs/config_parameters.pyi +7 -7
  218. {ob_metaflow_stubs-6.0.3.186.dist-info → ob_metaflow_stubs-6.0.3.188.dist-info}/METADATA +1 -1
  219. ob_metaflow_stubs-6.0.3.188.dist-info/RECORD +222 -0
  220. ob_metaflow_stubs-6.0.3.186.dist-info/RECORD +0 -216
  221. {ob_metaflow_stubs-6.0.3.186.dist-info → ob_metaflow_stubs-6.0.3.188.dist-info}/WHEEL +0 -0
  222. {ob_metaflow_stubs-6.0.3.186.dist-info → ob_metaflow_stubs-6.0.3.188.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
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.972156 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.032132 #
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
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.988372 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.048597 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.987224 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.047408 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.014750 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.075338 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.976125 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.036236 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,20 +1,20 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.980598 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.040594 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import tempfile
13
- import tarfile
14
12
  import datetime
15
- import metaflow.client.core
13
+ import tarfile
14
+ import tempfile
16
15
  import typing
17
16
  import metaflow.events
17
+ import metaflow.client.core
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
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.009719 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.070327 #
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.datastore.content_addressed_store
13
12
  import metaflow.exception
13
+ import metaflow.datastore.content_addressed_store
14
14
 
15
15
  from ..exception import MetaflowException as MetaflowException
16
16
 
metaflow-stubs/events.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.973122 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.033060 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.963643 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.023267 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,19 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.968177 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.028799 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
+ import metaflow.unbounded_foreach
12
13
  import metaflow.exception
14
+ import metaflow.flowspec
13
15
  import metaflow.datastore.inputs
14
- import metaflow.unbounded_foreach
15
16
  import typing
16
- import metaflow.flowspec
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-01T08:40:32.958176
1
+ 2.15.21.1+obcheckpoint(0.2.4);ob(v1) 2025-07-11T16:11:55.017667
@@ -1,16 +1,16 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.975458 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.035457 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
12
  import metaflow.parameters
13
+ import typing
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
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.962963 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.022565 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.967090 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.027690 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.994452 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.054656 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.993951 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.054156 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.metadata_provider.metadata
13
13
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.994177 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.054371 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.964775 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.024432 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.109750 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.173806 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -10,12 +10,12 @@ import typing
10
10
  if typing.TYPE_CHECKING:
11
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
12
12
  import metaflow.plugins.cards.component_serializer
13
- import typing
14
- import metaflow
15
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
16
- import metaflow.events
17
13
  import metaflow.metaflow_current
14
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
18
15
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
16
+ import metaflow
17
+ import typing
18
+ import metaflow.events
19
19
 
20
20
 
21
21
  TYPE_CHECKING: bool
@@ -228,6 +228,24 @@ class Current(object, metaclass=type):
228
228
  def graph(self):
229
229
  ...
230
230
  @property
231
+ def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
232
+ """
233
+ (only in the presence of the @card decorator)
234
+
235
+ The `@card` decorator makes the cards available through the `current.card`
236
+ object. If multiple `@card` decorators are present, you can add an `ID` to
237
+ distinguish between them using `@card(id=ID)` as the decorator. You will then
238
+ be able to access that specific card using `current.card[ID].
239
+
240
+ Methods available are `append` and `extend`
241
+
242
+ Returns
243
+ -------
244
+ CardComponentCollector
245
+ The or one of the cards attached to this step.
246
+ """
247
+ ...
248
+ @property
231
249
  def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
232
250
  """
233
251
  (only in the presence of the @huggingface_hub decorator)
@@ -255,21 +273,19 @@ class Current(object, metaclass=type):
255
273
  """
256
274
  ...
257
275
  @property
258
- def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
276
+ def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
259
277
  """
260
- (only in the presence of the @card decorator)
261
-
262
- The `@card` decorator makes the cards available through the `current.card`
263
- object. If multiple `@card` decorators are present, you can add an `ID` to
264
- distinguish between them using `@card(id=ID)` as the decorator. You will then
265
- be able to access that specific card using `current.card[ID].
278
+ (only in the presence of the @model decorator)
266
279
 
267
- Methods available are `append` and `extend`
280
+ The object used for loading / saving models.
281
+ `current.model` exposes a `save` method to save models and a `load` method to load models.
282
+ `current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
283
+ or models loaded via `current.model.load`.
268
284
 
269
285
  Returns
270
- -------
271
- CardComponentCollector
272
- The or one of the cards attached to this step.
286
+ ----------
287
+ ModelSerializer
288
+ The object used for loading / saving models.
273
289
  """
274
290
  ...
275
291
  @property
@@ -302,19 +318,16 @@ class Current(object, metaclass=type):
302
318
  """
303
319
  ...
304
320
  @property
305
- def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
321
+ def trigger(self) -> "metaflow.events.Trigger":
306
322
  """
307
- (only in the presence of the @model decorator)
323
+ (only in the presence of the @trigger_on_finish, or @trigger decorators)
308
324
 
309
- The object used for loading / saving models.
310
- `current.model` exposes a `save` method to save models and a `load` method to load models.
311
- `current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
312
- or models loaded via `current.model.load`.
325
+ Returns `Trigger` if the current run is triggered by an event
313
326
 
314
327
  Returns
315
- ----------
316
- ModelSerializer
317
- The object used for loading / saving models.
328
+ -------
329
+ Trigger
330
+ `Trigger` if triggered by an event
318
331
  """
319
332
  ...
320
333
  @property
@@ -384,19 +397,6 @@ class Current(object, metaclass=type):
384
397
  True if the flow is deployed with `--production`.
385
398
  """
386
399
  ...
387
- @property
388
- def trigger(self) -> "metaflow.events.Trigger":
389
- """
390
- (only in the presence of the @trigger, or @trigger_on_finish decorators)
391
-
392
- Returns `Trigger` if the current run is triggered by an event
393
-
394
- Returns
395
- -------
396
- Trigger
397
- `Trigger` if triggered by an event
398
- """
399
- ...
400
400
  ...
401
401
 
402
402
  current: Current
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.972349 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.032327 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.974385 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.034335 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.983340 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.043366 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.983399 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.043426 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.983455 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.043485 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.037475 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.099453 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.038049 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.100028 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.037521 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.099499 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.080043 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.143672 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.plugins.cards.card_modules.card
13
13
  import metaflow.plugins.cards.card_modules.components
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:32.983509 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.043557 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.038802 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.100781 #
5
5
  ######################################################################################################
6
6
 
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.038850 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.100832 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
+ import metaflow
9
10
  import threading
10
11
  import typing
11
- import metaflow
12
12
  if typing.TYPE_CHECKING:
13
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
14
13
  import threading
15
14
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
15
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
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
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.039222 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.101182 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.079005 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.142644 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
12
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
13
13
  import metaflow.datastore.datastore_storage
14
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
14
15
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
15
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
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
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.010016 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.070643 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.18.1+obcheckpoint(0.2.4);ob(v1) #
4
- # Generated on 2025-07-01T08:40:33.036608 #
3
+ # MF version: 2.15.21.1+obcheckpoint(0.2.4);ob(v1) #
4
+ # Generated on 2025-07-11T16:11:55.098606 #
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.checkpoints.checkpoint_storage
12
11
  import metaflow
13
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
14
12
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
13
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
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