ob-metaflow-stubs 6.0.10.7__py2.py3-none-any.whl → 6.0.10.9__py2.py3-none-any.whl

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.629616                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.005737                                                            #
 ######################################################################################################
 
 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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.665549                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.041814                                                            #
 ######################################################################################################
 
 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.checkpoints.core
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
-    import metaflow
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ......metaflow_current import current as current
 from ......exception import MetaflowException as MetaflowException

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

@@ -1,17 +1,17 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.631819                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.007963                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import os
     import metaflow
+    import metaflow.decorators
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from .exceptions import CheckpointException as CheckpointException

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.629837                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.005970                                                            #
 ######################################################################################################
 
 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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.594621                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:01.970340                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
+    import metaflow
 
 from ..datastructures import CheckpointArtifact as CheckpointArtifact
 from .constructors import load_checkpoint as load_checkpoint

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.668390                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.044688                                                            #
 ######################################################################################################
 
 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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.596274                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:01.971992                                                            #
 ######################################################################################################
 

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.596324                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:01.972040                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.669914                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.046244                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.668717                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.045039                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.777102                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.182618                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.630522                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.006644                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.671013                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.047333                                                            #
 ######################################################################################################
 
 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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.595612                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:01.971331                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.630159                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.006284                                                            #
 ######################################################################################################
 
 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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.632787                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.008963                                                            #
 ######################################################################################################
 

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.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.632836                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.009013                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -30,6 +30,32 @@ def download_model_from_huggingface(**kwargs):
 
 class HuggingfaceRegistry(object, metaclass=type):
     """
+    This object provides a thin, Metaflow-friendly layer over huggingface_hub's [snapshot_download](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/file_download#huggingface_hub.snapshot_download):
+    
+    - Snapshot references (persist-and-reuse): Use `current.huggingface_hub.snapshot_download(repo_id=..., ...)` to ensure a repo is available in the Metaflow datastore. If absent, it is downloaded once and saved; the call returns a reference dict you can store and load later (for example via `@model`).
+    
+    - On-demand local access (context manager):  Use `current.huggingface_hub.load(repo_id=..., [path=...], ...)` as a context manager to obtain a local filesystem path for immediate use. If the repo exists in the datastore, it is loaded from there; otherwise it is fetched from the Hugging Face Hub and then cached in the datastore. When `path` is omitted, a temporary directory is created and cleaned up automatically when the context exits. When `path` is provided, files are placed there and are not cleaned up by the context manager.
+    
+    Repos are cached in the datastore using the huggingface_hub.snapshot_download's arguments. The cache
+    key may include: `repo_id`, `repo_type`, `revision`, `ignore_patterns`,
+    and `allow_patterns` (see `cache_scope` for how keys are scoped).
+    
+    Examples
+    --------
+    ```python
+    # Snapshot reference:
+    ref = current.huggingface_hub.snapshot_download(
+        repo_id="google-bert/bert-base-uncased",
+        allow_patterns=["*.json"]
+    )
+    # Explicit Model Loading with Context manager:
+    
+    with current.huggingface_hub.load(
+        repo_id="google-bert/bert-base-uncased",
+        allow_patterns=["*.json"]
+    ) as local_path:
+        my_model = torch.load(os.path.join(local_path, "model.bin"))
+    ```
     """
     def __init__(self, logger):
         ...
@@ -139,75 +165,69 @@ class HuggingfaceHubDecorator(metaflow.mf_extensions.obcheckpoint.plugins.machin
     """
     Decorator that helps cache, version, and store models/datasets from the Hugging Face Hub.
     
-    > Examples
+    Examples
+    --------
     
-    **Usage: creating references to models from the Hugging Face Hub that may be loaded in downstream steps**
     ```python
-        @huggingface_hub
-        @step
-        def pull_model_from_huggingface(self):
-            # `current.huggingface_hub.snapshot_download` downloads the model from the Hugging Face Hub
-            # and saves it in the backend storage based on the model's `repo_id`. If there exists a model
-            # with the same `repo_id` in the backend storage, it will not download the model again. The return
-            # value of the function is a reference to the model in the backend storage.
-            # This reference can be used to load the model in the subsequent steps via `@model(load=["llama_model"])`
-    
-            self.model_id = "mistralai/Mistral-7B-Instruct-v0.1"
-            self.llama_model = current.huggingface_hub.snapshot_download(
-                repo_id=self.model_id,
-                allow_patterns=["*.safetensors", "*.json", "tokenizer.*"],
-            )
-            self.next(self.train)
-    ```
+    # **Usage: creating references to models from the Hugging Face Hub that may be loaded in downstream steps**
+    @huggingface_hub
+    @step
+    def pull_model_from_huggingface(self):
+        # `current.huggingface_hub.snapshot_download` downloads the model from the Hugging Face Hub
+        # and saves it in the backend storage based on the model's `repo_id`. If there exists a model
+        # with the same `repo_id` in the backend storage, it will not download the model again. The return
+        # value of the function is a reference to the model in the backend storage.
+        # This reference can be used to load the model in the subsequent steps via `@model(load=["llama_model"])`
     
-    **Usage: explicitly loading models at runtime from the Hugging Face Hub or from cache (from Metaflow's datastore)**
-    ```python
-        @huggingface_hub
-        @step
-        def run_training(self):
-            # Temporary directory (auto-cleaned on exit)
-            with current.huggingface_hub.load(
-                repo_id="google-bert/bert-base-uncased",
-                allow_patterns=["*.bin"],
-            ) as local_path:
-                # Use files under local_path
-                train_model(local_path)
-                ...
+        self.model_id = "mistralai/Mistral-7B-Instruct-v0.1"
+        self.llama_model = current.huggingface_hub.snapshot_download(
+            repo_id=self.model_id,
+            allow_patterns=["*.safetensors", "*.json", "tokenizer.*"],
+        )
+        self.next(self.train)
     
-    ```
+    # **Usage: explicitly loading models at runtime from the Hugging Face Hub or from cache (from Metaflow's datastore)**
+    @huggingface_hub
+    @step
+    def run_training(self):
+        # Temporary directory (auto-cleaned on exit)
+        with current.huggingface_hub.load(
+            repo_id="google-bert/bert-base-uncased",
+            allow_patterns=["*.bin"],
+        ) as local_path:
+            # Use files under local_path
+            train_model(local_path)
+            ...
     
-    **Usage: loading models directly from the Hugging Face Hub or from cache (from Metaflow's datastore)**
-    ```python
-        @huggingface_hub(load=["mistralai/Mistral-7B-Instruct-v0.1"])
-        @step
-        def pull_model_from_huggingface(self):
-            path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
-    ```
+    # **Usage: loading models directly from the Hugging Face Hub or from cache (from Metaflow's datastore)**
     
-    ```python
-        @huggingface_hub(load=[("mistralai/Mistral-7B-Instruct-v0.1", "/my-directory"), ("myorg/mistral-lora", "/my-lora-directory")])
-        @step
-        def finetune_model(self):
-            path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
-            # path_to_model will be /my-directory
-    ```
+    @huggingface_hub(load=["mistralai/Mistral-7B-Instruct-v0.1"])
+    @step
+    def pull_model_from_huggingface(self):
+        path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
     
-    ```python
-        # Takes all the arguments passed to `snapshot_download`
-        # except for `local_dir`
-        @huggingface_hub(load=[
-            {
-                "repo_id": "mistralai/Mistral-7B-Instruct-v0.1",
-            },
-            {
-                "repo_id": "myorg/mistral-lora",
-                "repo_type": "model",
-            },
-        ])
-        @step
-        def finetune_model(self):
-            path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
-            # path_to_model will be /my-directory
+    @huggingface_hub(load=[("mistralai/Mistral-7B-Instruct-v0.1", "/my-directory"), ("myorg/mistral-lora", "/my-lora-directory")])
+    @step
+    def finetune_model(self):
+        path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
+        # path_to_model will be /my-directory
+    
+    
+    # Takes all the arguments passed to `snapshot_download`
+    # except for `local_dir`
+    @huggingface_hub(load=[
+        {
+            "repo_id": "mistralai/Mistral-7B-Instruct-v0.1",
+        },
+        {
+            "repo_id": "myorg/mistral-lora",
+            "repo_type": "model",
+        },
+    ])
+    @step
+    def finetune_model(self):
+        path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
+        # path_to_model will be /my-directory
     ```
     
     Parameters
@@ -217,34 +237,30 @@ class HuggingfaceHubDecorator(metaflow.mf_extensions.obcheckpoint.plugins.machin
     
     cache_scope : str, optional
         The scope of the cache. Can be `checkpoint` / `flow` / `global`.
+            - `checkpoint` (default): All repos are stored like objects saved by `@checkpoint`.
+                i.e., the cached path is derived from the namespace, flow, step, and Metaflow foreach iteration.
+                Any repo downloaded under this scope will only be retrieved from the cache when the step runs under the same namespace in the same flow (at the same foreach index).
+    
+            - `flow`: All repos are cached under the flow, regardless of namespace.
+                i.e., the cached path is derived solely from the flow name.
+                When to use this mode: (1) Multiple users are executing the same flow and want shared access to the repos cached by the decorator. (2) Multiple versions of a flow are deployed, all needing access to the same repos cached by the decorator.
     
-        - `checkpoint` (default): All repos are stored like objects saved by `@checkpoint`.
-            i.e., the cached path is derived from the namespace, flow, step, and Metaflow foreach iteration.
-            Any repo downloaded under this scope will only be retrieved from the cache when the step runs under the same namespace in the same flow (at the same foreach index).
-    
-        - `flow`: All repos are cached under the flow, regardless of namespace.
-            i.e., the cached path is derived solely from the flow name.
-            When to use this mode:
-                - Multiple users are executing the same flow and want shared access to the repos cached by the decorator.
-                - Multiple versions of a flow are deployed, all needing access to the same repos cached by the decorator.
-    
-        - `global`: All repos are cached under a globally static path.
-            i.e., the base path of the cache is static and all repos are stored under it.
-            When to use this mode:
-                - All repos from the Hugging Face Hub need to be shared by users across all flow executions.
-    
-        Each caching scope comes with its own trade-offs:
-        - `checkpoint`:
-            - Has explicit control over when caches are populated (controlled by the same flow that has the `@huggingface_hub` decorator) but ends up hitting the Hugging Face Hub more often if there are many users/namespaces/steps.
-            - Since objects are written on a `namespace/flow/step` basis, the blast radius of a bad checkpoint is limited to a particular flow in a namespace.
-        - `flow`:
-            - Has less control over when caches are populated (can be written by any execution instance of a flow from any namespace) but results in more cache hits.
-            - The blast radius of a bad checkpoint is limited to all runs of a particular flow.
-            - It doesn't promote cache reuse across flows.
-        - `global`:
-            - Has no control over when caches are populated (can be written by any flow execution) but has the highest cache hit rate.
-            - It promotes cache reuse across flows.
-            - The blast radius of a bad checkpoint spans every flow that could be using a particular repo.
+            - `global`: All repos are cached under a globally static path.
+                i.e., the base path of the cache is static and all repos are stored under it.
+                When to use this mode:
+                    - All repos from the Hugging Face Hub need to be shared by users across all flow executions.
+            - Each caching scope comes with its own trade-offs:
+                - `checkpoint`:
+                    - Has explicit control over when caches are populated (controlled by the same flow that has the `@huggingface_hub` decorator) but ends up hitting the Hugging Face Hub more often if there are many users/namespaces/steps.
+                    - Since objects are written on a `namespace/flow/step` basis, the blast radius of a bad checkpoint is limited to a particular flow in a namespace.
+                - `flow`:
+                    - Has less control over when caches are populated (can be written by any execution instance of a flow from any namespace) but results in more cache hits.
+                    - The blast radius of a bad checkpoint is limited to all runs of a particular flow.
+                    - It doesn't promote cache reuse across flows.
+                - `global`:
+                    - Has no control over when caches are populated (can be written by any flow execution) but has the highest cache hit rate.
+                    - It promotes cache reuse across flows.
+                    - The blast radius of a bad checkpoint spans every flow that could be using a particular repo.
     
     load: Union[List[str], List[Tuple[Dict, str]], List[Tuple[str, str]], List[Dict], None]
         The list of repos (models/datasets) to load.
@@ -287,18 +303,15 @@ class HuggingfaceHubDecorator(metaflow.mf_extensions.obcheckpoint.plugins.machin
         key may include: `repo_id`, `repo_type`, `revision`, `ignore_patterns`,
         and `allow_patterns` (see `cache_scope` for how keys are scoped).
     
-        Examples
-        --------
-        Snapshot reference:
+        > Usage Styles
         ```python
+        # Snapshot reference:
         ref = current.huggingface_hub.snapshot_download(
             repo_id="google-bert/bert-base-uncased",
             allow_patterns=["*.json"]
         )
-        ```
     
-        Explicit Model Loading with Context manager:
-        ```python
+        # Explicit Model Loading with Context manager:
         with current.huggingface_hub.load(
             repo_id="google-bert/bert-base-uncased",
             allow_patterns=["*.json"]

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

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.633749                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.009942                                                            #
 ######################################################################################################
 

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

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.633796                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.009990                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.model_storage
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.model_storage
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
 
 from ..datastore.core import STORAGE_FORMATS as STORAGE_FORMATS

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.670836                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.047152                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.670609                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.046932                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
 import typing
 import metaflow
 if typing.TYPE_CHECKING:
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
     import metaflow.datastore.datastore_storage
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ..datastore.core import ObjectStorage as ObjectStorage
 from ..datastore.core import DatastoreInterface as DatastoreInterface

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

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.666101                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.042393                                                            #
 ######################################################################################################
 

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.666149                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.042441                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.671425                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.047697                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.671903                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.048199                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.672227                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.048513                                                            #
 ######################################################################################################
 

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.18.7.2+obcheckpoint(0.2.6);ob(v1)                                                    #
-# Generated on 2025-09-19T18:41:10.762816                                                            #
+# MF version: 2.18.7.5+obcheckpoint(0.2.7);ob(v1)                                                    #
+# Generated on 2025-09-23T00:18:02.163041                                                            #
 ######################################################################################################
 
 from __future__ import annotations