ob-metaflow-stubs 6.0.3.143rc0__py2.py3-none-any.whl → 6.0.3.145__py2.py3-none-any.whl

metaflow-stubs/cards.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.038869                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.083554                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/cli.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.053197                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.107799                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -22,8 +22,6 @@ from .user_configs.config_parameters import ConfigValue as ConfigValue
 from .cli_components.utils import cli as cli
 from .cli_components.utils import start as start
 
-DECOSPECS: str
-
 DEFAULT_DATASTORE: str
 
 DEFAULT_ENVIRONMENT: str
@@ -67,9 +65,6 @@ def echo_always(line, **kwargs):
 def logger(body = '', system_msg = False, head = '', bad = False, timestamp = True, nl = True):
     ...
 
-def config_merge_cb(ctx, param, value):
-    ...
-
 def print_metaflow_exception(ex):
     ...
 

metaflow-stubs/cli_components/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.051984                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.105819                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/cli_components/utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.077708                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.150411                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/client/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.042115                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.088689                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/client/core.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.046400                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.097052                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,10 +9,11 @@ from __future__ import annotations
 import typing
 import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow.events
+    import tarfile
     import metaflow.client.core
     import datetime
-    import tarfile
+    import metaflow.events
+    import tempfile
     import typing
 
 from ..metaflow_current import current as current
@@ -464,6 +465,50 @@ class MetaflowCode(object, metaclass=type):
             TarFile for everything in this code package
         """
         ...
+    def extract(self) -> tempfile.TemporaryDirectory:
+        """
+        Extracts the code package to a temporary directory.
+        
+        This creates a temporary directory containing all user code
+        files from the code package. The temporary directory is
+        automatically deleted when the returned TemporaryDirectory
+        object is garbage collected or when its cleanup() is called.
+        
+        To preserve the contents to a permanent location, use
+        os.replace() which performs a zero-copy move on the same
+        filesystem:
+        
+        ```python
+        with task.code.extract() as tmp_dir:
+            # Move contents to permanent location
+            for item in os.listdir(tmp_dir):
+                src = os.path.join(tmp_dir, item)
+                dst = os.path.join('/path/to/permanent/dir', item)
+                os.makedirs(os.path.dirname(dst), exist_ok=True)
+                os.replace(src, dst)  # Atomic move operation
+        ```
+        Returns
+        -------
+        TemporaryDirectory
+            A temporary directory containing the extracted code files.
+            The directory and its contents are automatically deleted when
+            this object is garbage collected.
+        """
+        ...
+    @property
+    def script_name(self) -> str:
+        """
+        Returns the filename of the Python script containing the FlowSpec.
+        
+        This is the main Python file that was used to execute the flow. For example,
+        if your flow is defined in 'myflow.py', this property will return 'myflow.py'.
+        
+        Returns
+        -------
+        str
+            Name of the Python file containing the FlowSpec
+        """
+        ...
     def __str__(self):
         ...
     ...
@@ -631,6 +676,28 @@ class Task(MetaflowObject, metaclass=type):
     def __init__(self, *args, **kwargs):
         ...
     @property
+    def parent_tasks(self) -> typing.Iterator["Task"]:
+        """
+        Yields all parent tasks of the current task if one exists.
+        
+        Yields
+        ------
+        Task
+            Parent task of the current task
+        """
+        ...
+    @property
+    def child_tasks(self) -> typing.Iterator["Task"]:
+        """
+        Yield all child tasks of the current task if one exists.
+        
+        Yields
+        ------
+        Task
+            Child task of the current task
+        """
+        ...
+    @property
     def metadata(self) -> typing.List[Metadata]:
         """
         Metadata events produced by this task across all attempts of the task
@@ -1090,6 +1157,28 @@ class Step(MetaflowObject, metaclass=type):
             Dictionary describing the environment
         """
         ...
+    @property
+    def parent_steps(self) -> typing.Iterator["Step"]:
+        """
+        Yields parent steps for the current step.
+        
+        Yields
+        ------
+        Step
+            Parent step
+        """
+        ...
+    @property
+    def child_steps(self) -> typing.Iterator["Step"]:
+        """
+        Yields child steps for the current step.
+        
+        Yields
+        ------
+        Step
+            Child step
+        """
+        ...
     ...
 
 class Run(MetaflowObject, metaclass=type):

metaflow-stubs/client/filecache.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.072816                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.141997                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/events.pyi

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.039466                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.084181                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.events
     import metaflow
+    import metaflow.events
 
 
 TYPE_CHECKING: bool

metaflow-stubs/exception.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.031118                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.076089                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/flowspec.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.035821                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.080723                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,10 +10,10 @@ import typing
 import metaflow
 if typing.TYPE_CHECKING:
     import metaflow.exception
+    import typing
+    import metaflow.datastore.inputs
     import metaflow.flowspec
     import metaflow.unbounded_foreach
-    import metaflow.datastore.inputs
-    import typing
 
 from . import parameters as parameters
 from .parameters import DelayedEvaluationParameter as DelayedEvaluationParameter

metaflow-stubs/generated_for.txt

@@ -1 +1 @@
-2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1) 2025-02-17T19:27:36.027826
+2.14.3.1+obcheckpoint(0.1.8);ob(v1) 2025-02-24T18:52:39.072458

metaflow-stubs/includefile.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.041828                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.088246                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
 import typing
 import metaflow
 if typing.TYPE_CHECKING:
+    import typing
     import metaflow.parameters
     import metaflow._vendor.click.types
-    import typing
 
 from ._vendor import click as click
 from .exception import MetaflowException as MetaflowException

metaflow-stubs/info_file.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.030377                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.075211                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata_provider/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.034705                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.079631                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata_provider/heartbeat.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.057009                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.115089                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata_provider/metadata.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.056529                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.114246                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import metaflow
 import typing
+import metaflow
 if typing.TYPE_CHECKING:
     import metaflow.metadata_provider.metadata
 
@@ -368,6 +368,30 @@ class MetadataProvider(object, metaclass=MetadataProviderMeta):
         Run tags after mutation operations
         """
         ...
+    @classmethod
+    def filter_tasks_by_metadata(cls, flow_name: str, run_id: str, step_name: str, field_name: str, pattern: str) -> typing.List[str]:
+        """
+        Filter tasks by metadata field and pattern, returning task pathspecs that match criteria.
+        
+        Parameters
+        ----------
+        flow_name : str
+            Flow name, that the run belongs to.
+        run_id: str
+            Run id, together with flow_id, that identifies the specific Run whose tasks to query
+        step_name: str
+            Step name to query tasks from
+        field_name: str
+            Metadata field name to query
+        pattern: str
+            Pattern to match in metadata field value
+        
+        Returns
+        -------
+        List[str]
+            List of task pathspecs that satisfy the query
+        """
+        ...
     def __init__(self, environment, flow, event_logger, monitor):
         ...
     ...

metaflow-stubs/metadata_provider/util.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.056751                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.114611                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metaflow_config.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.032215                                                            #
+# MF version: 2.14.3.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-24T18:52:39.077195                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -143,7 +143,7 @@ ARGO_WORKFLOWS_CAPTURE_ERROR_SCRIPT: None
 
 CONTACT_INFO: dict
 
-DECOSPECS: str
+DEFAULT_DECOSPECS: str
 
 ECS_S3_ACCESS_IAM_ROLE: None