ob-metaflow-stubs 6.0.3.154__py2.py3-none-any.whl → 6.0.3.155rc0__py2.py3-none-any.whl

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi

@@ -1,18 +1,18 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.946819                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.453263                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
-import threading
 import metaflow
+import threading
+import typing
 if typing.TYPE_CHECKING:
+    import threading
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
-    import threading
 
 from ...card_utils.deco_injection_mixin import CardDecoratorInjector as CardDecoratorInjector
 from ...card_utils.async_cards import CardRefresher as CardRefresher

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.947254                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.453861                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi

@@ -1,18 +1,18 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.983812                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.521318                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
     import metaflow.datastore.datastore_storage
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ..exceptions import KeyNotCompatibleWithObjectException as KeyNotCompatibleWithObjectException
 from ..utils.identity_utils import pathspec_hash as pathspec_hash
@@ -123,6 +123,8 @@ class CheckpointDatastore(metaflow.mf_extensions.obcheckpoint.plugins.machine_le
     @classmethod
     def decompose_key(cls, key) -> CheckpointsPathComponents:
         ...
+    def __str__(self):
+        ...
     def __init__(self, artifact_store: metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core.ObjectStorage, metadata_store: metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core.ObjectStorage, artifact_metadatastore: metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core.ObjectStorage):
         ...
     ...

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.916409                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.414761                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi

@@ -1,17 +1,17 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.943251                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.449162                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
+    import metaflow
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
-    import metaflow
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
 
 from ......metaflow_current import current as current
 from ......exception import MetaflowException as MetaflowException
@@ -22,6 +22,7 @@ from .checkpoint_storage import CheckpointDatastore as CheckpointDatastore
 from ..datastructures import CheckpointArtifact as CheckpointArtifact
 from ..datastore.task_utils import init_datastorage_object as init_datastorage_object
 from ..datastore.task_utils import resolve_storage_backend as resolve_task_storage_backend
+from ..datastore.task_utils import storage_backend_from_flow as storage_backend_from_flow
 
 TYPE_CHECKING: bool
 
@@ -45,6 +46,8 @@ class Checkpointer(object, metaclass=type):
         ...
     def load_metadata(self, version_id: int = None, name = 'mfchckpt') -> dict:
         ...
+    def artifact_id(self, name: str, version_id: int = None):
+        ...
     def save(self, path: str, metadata = {}, latest = True, name = 'mfchckpt', storage_format = 'files') -> metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures.CheckpointArtifact:
         ...
     ...
@@ -88,6 +91,9 @@ class ScopeResolver(object, metaclass=type):
         ...
     ...
 
+def warning_message(message, logger = None, ts = False, prefix = '[@checkpoint][warning]'):
+    ...
+
 class ReadResolver(object, metaclass=type):
     """
     Responsible for instantiating the `CheckpointDatastore` during read operations
@@ -102,7 +108,7 @@ class ReadResolver(object, metaclass=type):
         """
         ...
     @classmethod
-    def from_key_and_pathspec(cls, pathspec, checkpoint_key):
+    def from_checkpoint(cls, checkpoint: metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures.CheckpointArtifact):
         """
         """
         ...

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi

@@ -1,18 +1,18 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.918537                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.416891                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
-    import os
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
-    import metaflow.decorators
     import metaflow
+    import metaflow.decorators
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
+    import os
 
 from .exceptions import CheckpointException as CheckpointException
 from ..utils import flowspec_utils as flowspec_utils
@@ -28,6 +28,7 @@ from .constructors import load_checkpoint as load_checkpoint
 from .core import ScopeResolver as ScopeResolver
 from .core import CheckpointLoadPolicy as CheckpointLoadPolicy
 from ..datastructures import CheckpointArtifact as CheckpointArtifact
+from ..datastore.decorator import set_datastore_context as set_datastore_context
 from .final_api import Checkpoint as Checkpoint
 
 DEFAULT_NAME: str

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.916640                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.414980                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.893051                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.390393                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
     import metaflow
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ..datastructures import CheckpointArtifact as CheckpointArtifact
 from .constructors import load_checkpoint as load_checkpoint
@@ -63,6 +63,8 @@ class Checkpoint(object, metaclass=type):
             If `files`, saves directory directly to the datastore.
         """
         ...
+    def generate_key(self, name: str, version_id: int = None):
+        ...
     def __enter__(self):
         ...
     def __exit__(self, exc_type, exc_val, exc_tb):

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.947530                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.454251                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.917233                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.392105                                                            #
 ######################################################################################################
 

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi

@@ -0,0 +1,95 @@
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.392151                                                            #
+######################################################################################################
+
+from __future__ import annotations
+
+import metaflow
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.context
+    import metaflow.exception
+    import metaflow
+
+from ......exception import MetaflowException as MetaflowException
+
+TYPE_CHECKING: bool
+
+ARTIFACT_STORE_CONFIG_ENV_VAR: str
+
+def will_switch_context(func):
+    """
+    A decorator that wraps methods in ArtifactStoreContext.
+    It takes the return value of the decorated method and passes it to
+    self.switch_context(), then returns None.
+    """
+    ...
+
+class DatastoreContextValidation(tuple, metaclass=type):
+    """
+    DatastoreContextInfo(is_valid, needs_external_context, context_mismatch)
+    """
+    @staticmethod
+    def __new__(_cls, is_valid, needs_external_context, context_mismatch):
+        """
+        Create new instance of DatastoreContextInfo(is_valid, needs_external_context, context_mismatch)
+        """
+        ...
+    def __repr__(self):
+        """
+        Return a nicely formatted representation string
+        """
+        ...
+    def __getnewargs__(self):
+        """
+        Return self as a plain tuple.  Used by copy and pickle.
+        """
+        ...
+    ...
+
+class UnresolvableDatastoreException(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+class ArtifactStoreContext(object, metaclass=type):
+    """
+    This class will act as a singleton to switch the datastore so that
+    Any checkpoint operations will be done in the correct datastore. This
+    can be a useful way of short-circuting a datastore switch between runtime
+    And post-runtime retrieval operations.
+    """
+    @property
+    def MF_DATASTORES(self):
+        ...
+    def __init__(self):
+        ...
+    def flow_init_context(self, *args, **kwargs):
+        ...
+    def switch_context(self, context):
+        ...
+    def get(self):
+        ...
+    def context_from_run(self, *args, **kwargs):
+        ...
+    def context_from_task(self, *args, **kwargs):
+        ...
+    def current_context_matches_task_metadata(self, task: "metaflow.Task"):
+        ...
+    def to_task_metadata(self):
+        ...
+    def default(self):
+        ...
+    ...
+
+datastore_context: ArtifactStoreContext
+
+def artifact_store_from(run = None, task = None, config = None):
+    """
+    This context manager can be used to switch the artifact store
+    for a block of code. This is useful when users maybe accessing
+    checkpoints/models from a different datastore using the
+    `@with_artifact_store` decorator.
+    """
+    ...
+

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.944494                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.457016                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -105,9 +105,6 @@ class ObjectStorage(object, metaclass=type):
         ...
     def set_full_prefix(self, root_prefix):
         ...
-    @staticmethod
-    def resolve_root(storage_type):
-        ...
     def full_base_url(self, prefix = None):
         ...
     def create_key_name(self, *args):
@@ -142,6 +139,8 @@ class ObjectStorage(object, metaclass=type):
         List all objects in the datastore's `keys` index.
         """
         ...
+    def __str__(self) -> str:
+        ...
     ...
 
 class DatastoreInterface(object, metaclass=type):
@@ -183,6 +182,3 @@ class DatastoreInterface(object, metaclass=type):
         ...
     ...
 
-def resolve_root(datastore_type):
-    ...
-

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi

@@ -0,0 +1,135 @@
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.454787                                                            #
+######################################################################################################
+
+from __future__ import annotations
+
+import metaflow
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.decorators
+
+from ......metadata_provider.metadata import MetaDatum as MetaDatum
+from ......exception import MetaflowException as MetaflowException
+
+class ArtifactStoreFlowDecorator(metaflow.decorators.FlowDecorator, metaclass=type):
+    """
+    Allows setting external datastores to save data for the
+    `@checkpoint`/`@model`/`@huggingface_hub` decorators.
+    
+    This decorator is useful when users wish to save data to a different datastore
+    than what is configured in Metaflow. This can be for variety of reasons:
+    
+    1. Data security: The objects needs to be stored in a bucket (object storage) that is not accessible by other flows.
+    2. Data Locality: The location where the task is executing is not located in the same region as the datastore.
+        - Example: Metaflow datastore lives in US East, but the task is executing in Finland datacenters.
+    3. Data Lifecycle Policies: The objects need to be archived / managed separately from the Metaflow managed objects.
+        - Example: Flow is training very large models that need to be stored separately and will be deleted more aggressively than the Metaflow managed objects.
+    
+    Usage:
+    ----------
+    
+    - Using a custom IAM role to access the datastore.
+    
+        ```python
+        @with_artifact_store(
+            type="s3",
+            config=lambda: {
+                "root": "s3://my-bucket-foo/path/to/root",
+                "role_arn": ROLE,
+            },
+        )
+        class MyFlow(FlowSpec):
+    
+            @checkpoint
+            @step
+            def start(self):
+                with open("my_file.txt", "w") as f:
+                    f.write("Hello, World!")
+                self.external_bucket_checkpoint = current.checkpoint.save("my_file.txt")
+                self.next(self.end)
+    
+        ```
+    
+    - Using credentials to access the s3-compatible datastore.
+    
+        ```python
+        @with_artifact_store(
+            type="s3",
+            config=lambda: {
+                "root": "s3://my-bucket-foo/path/to/root",
+                "client_params": {
+                    "aws_access_key_id": os.environ.get("MY_CUSTOM_ACCESS_KEY"),
+                    "aws_secret_access_key": os.environ.get("MY_CUSTOM_SECRET_KEY"),
+                },
+            },
+        )
+        class MyFlow(FlowSpec):
+    
+            @checkpoint
+            @step
+            def start(self):
+                with open("my_file.txt", "w") as f:
+                    f.write("Hello, World!")
+                self.external_bucket_checkpoint = current.checkpoint.save("my_file.txt")
+                self.next(self.end)
+    
+        ```
+    
+    - Accessing objects stored in external datastores after task execution.
+    
+        ```python
+        run = Run("CheckpointsTestsFlow/8992")
+        with artifact_store_from(run=run, config={
+            "client_params": {
+                "aws_access_key_id": os.environ.get("MY_CUSTOM_ACCESS_KEY"),
+                "aws_secret_access_key": os.environ.get("MY_CUSTOM_SECRET_KEY"),
+            },
+        }):
+            with Checkpoint() as cp:
+                latest = cp.list(
+                    task=run["start"].task
+                )[0]
+                print(latest)
+                cp.load(
+                    latest,
+                    "test-checkpoints"
+                )
+    
+        task = Task("TorchTuneFlow/8484/train/53673")
+        with artifact_store_from(run=run, config={
+            "client_params": {
+                "aws_access_key_id": os.environ.get("MY_CUSTOM_ACCESS_KEY"),
+                "aws_secret_access_key": os.environ.get("MY_CUSTOM_SECRET_KEY"),
+            },
+        }):
+            load_model(
+                task.data.model_ref,
+                "test-models"
+            )
+        ```
+    Parameters:
+    ----------
+    
+    type: str
+        The type of the datastore. Can be one of 's3', 'gcs', 'azure' or any other supported metaflow Datastore.
+    
+    config: dict or Callable
+        Dictionary of configuration options for the datastore. The following keys are required:
+        - root: The root path in the datastore where the data will be saved. (needs to be in the format expected by the datastore)
+            - example: 's3://bucket-name/path/to/root'
+            - example: 'gs://bucket-name/path/to/root'
+            - example: 'https://myblockacc.blob.core.windows.net/metaflow/'
+        - role_arn (optional): AWS IAM role to access s3 bucket (only when `type` is 's3')
+        - session_vars (optional): AWS session variables to access s3 bucket (only when `type` is 's3')
+        - client_params (optional): AWS client parameters to access s3 bucket (only when `type` is 's3')
+    """
+    def flow_init(self, flow, graph, environment, flow_datastore, metadata, logger, echo, options):
+        ...
+    ...
+
+def set_datastore_context(flow, metadata, run_id, step_name, task_id, retry_count):
+    ...
+

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.999085                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.542883                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi

@@ -1,19 +1,20 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.917283                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.415634                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
     import metaflow
 
+from ......metaflow_current import current as current
 from ......exception import MetaflowException as MetaflowException
-from .core import resolve_root as resolve_root
+from .context import datastore_context as datastore_context
 
 TYPE_CHECKING: bool
 
@@ -24,5 +25,32 @@ def init_datastorage_object():
     ...
 
 def resolve_storage_backend(pathspec: typing.Union[str, "metaflow.Task"] = None):
+    """
+    This function is ONLY called when the user is calling `Checkpoint.list`.
+    What happens when users call list:
+        1. The task has completed so the list call is using the `_task_checkpoints` data-artifact to list the checkpoints.
+        2. The task is still running or has crashed. This means that in order to list the checkpoints, we need access to the datastore.
+    
+    For case `1`, this code path wont even be called because the data-artifact is already a separate object.
+    For case `2` is where this code path is important. But since we now expose a `artifact_store_from` context manager, we
+    know pre-hand what the datastore needs to be and the datastore context has already been switched.
+    
+    SO in turn, one can make the argument that this function is not needed and we can just have a
+    users do a `Checkpoint.list` under the context manager if they need to access objects in a different datastore.
+    
+    This is a respectible pattern since directly reading the task metadata and trying to do a list call is not a good pattern
+    since the creds of metaflow default datastore might not be the same as the datastore the user wants to access.
+    
+    There is a function that can help verify if the datastore set in the task metadata
+    is the same as the default datastore. If it is not, then we should shout warning messages
+    to the user. In case the permissions are the same, nothing wrong happens, if they are not then
+    user will have some hint in the logs to help them figure out the issue.
+    """
+    ...
+
+class FlowNotRunningException(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+def storage_backend_from_flow(flow: "metaflow.FlowSpec"):
     ...
 

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.948626                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.459044                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.894076                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.391435                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.916954                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.415291                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.919341                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.417687                                                            #
 ######################################################################################################
 

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1)                                                    #
-# Generated on 2025-04-10T22:13:14.919388                                                            #
+# MF version: 2.15.7.2+obcheckpoint(0.2.0);ob(v1)                                                    #
+# Generated on 2025-04-14T19:55:43.417772                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
 
 from ..checkpoints.decorator import CheckpointDecorator as CheckpointDecorator
 from ..checkpoints.decorator import CurrentCheckpointer as CurrentCheckpointer