metaflow-stubs 2.14.0__py2.py3-none-any.whl → 2.14.1__py2.py3-none-any.whl

metaflow-stubs/cards.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.271649                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.322643                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/cli.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.279981                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.331373                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/cli_components/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.278804                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.330070                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/cli_components/utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.299610                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.351358                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/client/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.271904                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.322897                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/client/core.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.276355                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.327577                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,10 +9,11 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
+    import datetime
+    import tempfile
+    import metaflow.events
     import tarfile
     import typing
-    import metaflow.events
-    import datetime
     import metaflow.client.core
 
 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                                                                                 #
-# Generated on 2025-02-11T19:31:25.299240                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.350987                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/events.pyi

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.265829                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.316785                                                            #
 ######################################################################################################
 
 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                                                                                 #
-# Generated on 2025-02-11T19:31:25.257535                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.308464                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/flowspec.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.262424                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.313366                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,10 +9,10 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.unbounded_foreach
-    import metaflow.datastore.inputs
+    import metaflow.exception
     import typing
+    import metaflow.datastore.inputs
     import metaflow.flowspec
 
 from . import parameters as parameters

metaflow-stubs/generated_for.txt

@@ -1 +1 @@
-2.14.0 2025-02-11T19:31:25.255952
+2.14.1 2025-02-21T00:53:08.306883

metaflow-stubs/includefile.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.271440                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.322427                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/info_file.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.256642                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.307589                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata_provider/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.261305                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.312256                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata_provider/heartbeat.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.282696                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.334152                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata_provider/metadata.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.282183                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.333643                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -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                                                                                 #
-# Generated on 2025-02-11T19:31:25.282425                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.333874                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/metaflow_config.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.258707                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.309633                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -247,8 +247,6 @@ CONDA_PACKAGE_GSROOT: None
 
 CONDA_DEPENDENCY_RESOLVER: str
 
-CONDA_USE_FAST_INIT: bool
-
 ESCAPE_HATCH_WARNING: bool
 
 DEBUG_OPTIONS: list

metaflow-stubs/metaflow_current.pyi

@@ -1,17 +1,17 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.350652                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.403062                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import typing
-    import metaflow.events
-    import metaflow
     import metaflow.metaflow_current
+    import metaflow
+    import metaflow.events
+    import typing
     import metaflow.plugins.cards.component_serializer
 
 
@@ -225,6 +225,24 @@ class Current(object, metaclass=type):
     def graph(self):
         ...
     @property
+    def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
+        """
+        (only in the presence of the @card decorator)
+        
+        The `@card` decorator makes the cards available through the `current.card`
+        object. If multiple `@card` decorators are present, you can add an `ID` to
+        distinguish between them using `@card(id=ID)` as the decorator. You will then
+        be able to access that specific card using `current.card[ID].
+        
+        Methods available are `append` and `extend`
+        
+        Returns
+        -------
+        CardComponentCollector
+            The or one of the cards attached to this step.
+        """
+        ...
+    @property
     def parallel(self) -> "metaflow.metaflow_current.Parallel":
         """
         (only in the presence of the @parallel decorator)
@@ -254,21 +272,16 @@ class Current(object, metaclass=type):
         """
         ...
     @property
-    def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
+    def trigger(self) -> "metaflow.events.Trigger":
         """
-        (only in the presence of the @card decorator)
-        
-        The `@card` decorator makes the cards available through the `current.card`
-        object. If multiple `@card` decorators are present, you can add an `ID` to
-        distinguish between them using `@card(id=ID)` as the decorator. You will then
-        be able to access that specific card using `current.card[ID].
+        (only in the presence of the @trigger, or @trigger_on_finish decorators)
         
-        Methods available are `append` and `extend`
+        Returns `Trigger` if the current run is triggered by an event
         
         Returns
         -------
-        CardComponentCollector
-            The or one of the cards attached to this step.
+        Trigger
+            `Trigger` if triggered by an event
         """
         ...
     @property
@@ -338,19 +351,6 @@ class Current(object, metaclass=type):
             True if the flow is deployed with `--production`.
         """
         ...
-    @property
-    def trigger(self) -> "metaflow.events.Trigger":
-        """
-        (only in the presence of the @trigger_on_finish, or @trigger decorators)
-        
-        Returns `Trigger` if the current run is triggered by an event
-        
-        Returns
-        -------
-        Trigger
-            `Trigger` if triggered by an event
-        """
-        ...
     ...
 
 current: Current

metaflow-stubs/multicore_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.259151                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.310084                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/parameters.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.260519                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.311467                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.266919                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.317912                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,23 +9,23 @@ from __future__ import annotations
 
 from . import test_unbounded_foreach_decorator as test_unbounded_foreach_decorator
 from .test_unbounded_foreach_decorator import InternalTestUnboundedForeachInput as InternalTestUnboundedForeachInput
-from . import retry_decorator as retry_decorator
 from . import pypi as pypi
-from . import parallel_decorator as parallel_decorator
+from . import cards as cards
 from . import timeout_decorator as timeout_decorator
-from . import catch_decorator as catch_decorator
-from . import resources_decorator as resources_decorator
-from . import argo as argo
-from . import environment_decorator as environment_decorator
 from . import secrets as secrets
 from . import kubernetes as kubernetes
+from . import resources_decorator as resources_decorator
 from . import aws as aws
+from . import argo as argo
+from . import catch_decorator as catch_decorator
 from . import datatools as datatools
 from . import frameworks as frameworks
+from . import parallel_decorator as parallel_decorator
 from . import airflow as airflow
-from . import cards as cards
-from . import project_decorator as project_decorator
+from . import environment_decorator as environment_decorator
+from . import retry_decorator as retry_decorator
 from . import events_decorator as events_decorator
+from . import project_decorator as project_decorator
 from . import gcp as gcp
 from . import storage_executor as storage_executor
 from . import azure as azure

metaflow-stubs/plugins/airflow/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.289505                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.341031                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/airflow_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.313188                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.368923                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/airflow_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.312889                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.368622                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/exception.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.313385                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.369103                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.313614                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.369354                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.328247                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.382496                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.328513                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.382756                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0                                                                                 #
-# Generated on 2025-02-11T19:31:25.328725                                                            #
+# MF version: 2.14.1                                                                                 #
+# Generated on 2025-02-21T00:53:08.382972                                                            #
 ######################################################################################################
 
 from __future__ import annotations