ob-metaflow-stubs 6.0.3.138__py2.py3-none-any.whl → 6.0.3.140__py2.py3-none-any.whl

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi

@@ -1,16 +1,16 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.655556                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.169302                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.cards.card_modules.card
     import metaflow.plugins.cards.card_modules.components
+    import metaflow.plugins.cards.card_modules.card
 
 from ......plugins.cards.card_modules.components import VegaChart as VegaChart
 from ......plugins.cards.card_modules.card import MetaflowCardComponent as MetaflowCardComponent

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

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.560461                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.072764                                                            #
 ######################################################################################################
 

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

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.613884                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.125705                                                            #
 ######################################################################################################
 

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

@@ -1,14 +1,14 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.613931                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.125769                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
-import typing
 import threading
+import typing
+import metaflow
 if typing.TYPE_CHECKING:
     import threading
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.614298                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.126276                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,17 +1,17 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.654293                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.167973                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
+    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.checkpoints.checkpoint_storage
-    import metaflow.datastore.datastore_storage
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ..exceptions import KeyNotCompatibleWithObjectException as KeyNotCompatibleWithObjectException
@@ -118,7 +118,7 @@ class CheckpointDatastore(metaflow.mf_extensions.obcheckpoint.plugins.machine_le
         ...
     def load_metadata(self, attempt, version_id, name = 'mfchckpt') -> dict:
         ...
-    def list(self, name = None, attempt = None, within_task = True):
+    def list(self, name: typing.Optional[str] = None, attempt: typing.Optional[int] = None, within_task: typing.Optional[bool] = True):
         ...
     @classmethod
     def decompose_key(cls, key) -> CheckpointsPathComponents:

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.583599                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.095588                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.610192                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.121933                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
     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.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ......metaflow_current import current as current
@@ -94,6 +94,9 @@ class ReadResolver(object, metaclass=type):
     based on different context's.
     """
     @classmethod
+    def from_pathspec(cls, pathspec):
+        ...
+    @classmethod
     def from_key(cls, checkpoint_key):
         """
         """

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

@@ -1,16 +1,16 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.585683                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.097714                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow
     import metaflow.decorators
+    import metaflow
     import os
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
@@ -97,7 +97,57 @@ class CurrentCheckpointer(object, metaclass=type):
             If `files`, saves directory directly to the datastore.
         """
         ...
-    def list(self, name: typing.Optional[str] = None, task: typing.Union["metaflow.Task", str, None] = None, attempt: typing.Optional[int] = None, within_task: bool = True):
+    def list(self, name: typing.Optional[str] = None, task: typing.Union["metaflow.Task", str, None] = None, attempt: typing.Optional[int] = None, full_namespace: bool = False) -> typing.List[typing.Dict]:
+        """
+        lists the checkpoints in the current task or the specified task.
+        
+        When users call `list` without any arguments, it will list all the checkpoints in the currently executing
+        task (this includes all attempts). If the `list` method is called without any arguments outside a Metaflow Task execution context,
+        it will raise an exception. Users can also call `list` with `attempt` argument to list all checkpoints within a
+        the specific attempt of the currently executing task.
+        
+        When a `task` argument is provided, the `list` method will return all the checkpoints
+        for a task's latest attempt unless a specific attempt number is set in the `attempt` argument.
+        If the `Task` object contains a `DataArtifact` with all the previous checkpoints, then the `list` method will return
+        all the checkpoints from the data artifact. If for some reason the DataArtifact is not written, then the `list` method will
+        return all checkpoints directly from the checkpoint's datastore.
+        
+        Usage:
+        ------
+        
+        ```python
+        current.checkpoint.list(name="best") # lists checkpoints in the current task with the name "best"
+        current.checkpoint.list( # Identical as the above one but lists checkpoints from the specified task with the name "best"
+            task="anotherflow/somerunid/somestep/sometask",
+            name="best"
+        )
+        current.checkpoint.list() # lists **all** the checkpoints in the current task (including the ones from all attempts)
+        ```
+        
+        Parameters
+        ----------
+        
+        name : Optional[str], default: None
+            Filter checkpoints by name.
+        
+        task : Optional[Union["metaflow.Task", str]], default: None
+            The task to list checkpoints from. Can be either a `Task` object or a task pathspec string.
+            If None, lists checkpoints for the current task.
+            Raises an exception if task is not provided when called outside a Metaflow Task execution context.
+        
+        attempt : Optional[Union[int, str]], default: None
+            Filter checkpoints by attempt.
+            If `task` is not None and `attempt` is None, then it will load the task's latest attempt
+        
+        full_namespace : bool, default: False
+            If True, lists checkpoints from the full namespace.
+            Only allowed during a Metaflow Task execution context.
+            Raises an exception if `full_namespace` is set to True when called outside a Metaflow Task execution context.
+        
+        Returns
+        -------
+        List[Dict]
+        """
         ...
     def cleanup(self):
         ...

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.583815                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.095798                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.560501                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.072803                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -14,6 +14,7 @@ if typing.TYPE_CHECKING:
 from ..datastructures import CheckpointArtifact as CheckpointArtifact
 from .constructors import load_checkpoint as load_checkpoint
 from .exceptions import CheckpointNotAvailableException as CheckpointNotAvailableException
+from .exceptions import CheckpointException as CheckpointException
 
 TYPE_CHECKING: bool
 
@@ -31,7 +32,7 @@ class Checkpoint(object, metaclass=type):
     @property
     def directory(self):
         ...
-    def save(self, path = None, metadata = None, latest = True, name = 'mfchckpt', storage_format = 'files'):
+    def save(self, path = None, metadata = None, latest = True, name = 'mfchckpt', storage_format = 'files') -> typing.Dict:
         """
         Saves the checkpoint to the datastore
         
@@ -66,10 +67,20 @@ class Checkpoint(object, metaclass=type):
         ...
     def __exit__(self, exc_type, exc_val, exc_tb):
         ...
-    def list(self, name: typing.Optional[str] = None, task: typing.Union["metaflow.Task", str, None] = None, attempt: typing.Union[int, str, None] = None, as_dict: bool = True, within_task: bool = True) -> typing.Iterable[typing.Union[typing.Dict, metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures.CheckpointArtifact]]:
+    def list(self, name: typing.Optional[str] = None, task: typing.Union["metaflow.Task", str, None] = None, attempt: typing.Union[int, str, None] = None, full_namespace: bool = False, as_dict: bool = True) -> typing.List[typing.Union[typing.Dict, metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures.CheckpointArtifact]]:
         """
-        lists the checkpoints in the datastore based on the Task.
-        It will always be task scoped.
+        lists the checkpoints in the current task or the specified task.
+        
+        When users call `list` without any arguments, it will list all the checkpoints in the currently executing
+        task (this includes all attempts). If the `list` method is called without any arguments outside a Metaflow Task execution context,
+        it will raise an exception. Users can also call `list` with `attempt` argument to list all checkpoints within a
+        the specific attempt of the currently executing task.
+        
+        When a `task` argument is provided, the `list` method will return all the checkpoints
+        for a task's latest attempt unless a specific attempt number is set in the `attempt` argument.
+        If the `Task` object contains a `DataArtifact` with all the previous checkpoints, then the `list` method will return
+        all the checkpoints from the data artifact. If for some reason the DataArtifact is not written, then the `list` method will
+        return all checkpoints directly from the checkpoint's datastore.
         
         Usage:
         ------
@@ -78,19 +89,33 @@ class Checkpoint(object, metaclass=type):
         
         Checkpoint().list(name="best") # lists checkpoints in the current task with the name "best"
         Checkpoint().list(task="anotherflow/somerunid/somestep/sometask", name="best") # Identical as the above one but
-        Checkpoint().list() # lists all the checkpoints in the current task
+        Checkpoint().list() # lists **all** the checkpoints in the current task (including the ones from all attempts)
         
         ```
         
         Parameters
         ----------
         
-        - `name`:
-            - name of the checkpoint to filter for
-        - `task`:
-            - Task object outside the one that is currently set in the `Checkpoint` object; Can be a pathspec string.
-        - `attempt`:
-            - attempt number of the task (optional filter. If none, then lists all checkpoints from all attempts)
+        name : Optional[str], default: None
+            Filter checkpoints by name.
+        
+        task : Optional[Union["metaflow.Task", str]], default: None
+            The task to list checkpoints from. Can be either a `Task` object or a task pathspec string.
+            If None, lists checkpoints for the current task.
+            Raises an exception if task is not provided when called outside a Metaflow Task execution context.
+        
+        attempt : Optional[Union[int, str]], default: None
+            Filter checkpoints by attempt.
+            If `task` is not None and `attempt` is None, then it will load the task's latest attempt
+        
+        full_namespace : bool, default: False
+            If True, lists checkpoints from the full namespace.
+            Only allowed during a Metaflow Task execution context.
+            Raises an exception if `full_namespace` is set to True when called outside a Metaflow Task execution context.
+        
+        Returns
+        -------
+        List[Dict]
         """
         ...
     def load(self, reference: typing.Union[str, typing.Dict, metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures.CheckpointArtifact], path: typing.Optional[str] = None):

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.614576                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.126582                                                            #
 ######################################################################################################
 
 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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.584423                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.096398                                                            #
 ######################################################################################################
 

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

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.611558                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.123260                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    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.datastore.core
 
 from ......exception import MetaflowException as MetaflowException
 from ..utils.tar_utils import create_tarball_on_disk as create_tarball_on_disk

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.670502                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.183295                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,16 +1,16 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.584470                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.096444                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow
     import metaflow.exception
+    import metaflow
 
 from ......exception import MetaflowException as MetaflowException
 from .core import resolve_root as resolve_root

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.615707                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.127694                                                            #
 ######################################################################################################
 
 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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.561471                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.073779                                                            #
 ######################################################################################################
 
 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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.584133                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.096114                                                            #
 ######################################################################################################
 
 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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.586569                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.098544                                                            #
 ######################################################################################################
 

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.586622                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.098592                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.587517                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.099491                                                            #
 ######################################################################################################
 

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.587592                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.099541                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ 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.core
+    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ..datastore.core import STORAGE_FORMATS as STORAGE_FORMATS
 from ..exceptions import KeyNotFoundError as KeyNotFoundError

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.615498                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.127512                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,16 +1,16 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.615283                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.127278                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
-    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.datastore.core
     import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
 
 from ..datastore.core import ObjectStorage as ObjectStorage

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

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.612143                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.123849                                                            #
 ######################################################################################################
 

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.612194                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.123898                                                            #
 ######################################################################################################
 
 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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.616083                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.128070                                                            #
 ######################################################################################################
 
 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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.616575                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.128586                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow
     import metaflow.exception
+    import metaflow
 
 from ......exception import MetaflowException as MetaflowException
 

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.616914                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.128918                                                            #
 ######################################################################################################
 

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.13.9.2+obcheckpoint(0.1.7);ob(v1)                                                    #
-# Generated on 2025-02-03T21:17:39.655838                                                            #
+# MF version: 2.14.0.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-11T20:37:27.169586                                                            #
 ######################################################################################################
 
 from __future__ import annotations