metaflow-stubs 2.17.1__py2.py3-none-any.whl → 2.17.2__py2.py3-none-any.whl

metaflow-stubs/plugins/uv/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.854681                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.520878                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/uv/uv_environment.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.888416                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.552027                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/pylint_wrapper.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.835798                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.497743                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/runner/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.821107                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.482701                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/runner/deployer.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.930564                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.591559                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,12 +9,12 @@ from __future__ import annotations
 import typing
 import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.argo.argo_workflows_deployer
     import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
-    import metaflow
     import metaflow.runner.deployer
-    import metaflow.plugins.aws.step_functions.step_functions_deployer
     import metaflow.plugins.argo.argo_workflows_deployer_objects
+    import metaflow.plugins.aws.step_functions.step_functions_deployer
+    import metaflow.plugins.argo.argo_workflows_deployer
+    import metaflow
 
 from ..exception import MetaflowNotFound as MetaflowNotFound
 
@@ -58,24 +58,24 @@ class Deployer(object, metaclass=DeployerMeta):
     """
     def __init__(self, flow_file: str, show_output: bool = True, profile: typing.Optional[str] = None, env: typing.Optional[typing.Dict] = None, cwd: typing.Optional[str] = None, file_read_timeout: int = 3600, **kwargs):
         ...
-    def argo_workflows(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer.ArgoWorkflowsDeployer":
+    def step_functions(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.aws.step_functions.step_functions_deployer.StepFunctionsDeployer":
         """
-        Deployer implementation for Argo Workflows.
+        Deployer implementation for AWS Step Functions.
         
         Parameters
         ----------
         name : str, optional, default None
-            Argo workflow name. The flow name is used instead if this option is not specified.
+            State Machine name. The flow name is used instead if this option is not specified.
         """
         ...
-    def step_functions(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.aws.step_functions.step_functions_deployer.StepFunctionsDeployer":
+    def argo_workflows(self, *, name: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer.ArgoWorkflowsDeployer":
         """
-        Deployer implementation for AWS Step Functions.
+        Deployer implementation for Argo Workflows.
         
         Parameters
         ----------
         name : str, optional, default None
-            State Machine name. The flow name is used instead if this option is not specified.
+            Argo workflow name. The flow name is used instead if this option is not specified.
         """
         ...
     ...
@@ -206,58 +206,58 @@ class DeployedFlow(object, metaclass=DeployedFlowMeta):
         """
         ...
     @classmethod
-    def from_argo_workflows(cls, *, identifier: str, metadata: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer_objects.ArgoWorkflowsDeployedFlow":
+    def from_step_functions(cls) -> "metaflow.plugins.aws.step_functions.step_functions_deployer_objects.StepFunctionsDeployedFlow":
         """
-        Retrieves a `ArgoWorkflowsDeployedFlow` object from an identifier and optional
-        metadata.
-        
+        This method is not currently implemented for Step Functions.
         
+        Raises
+        ------
+        NotImplementedError
+            This method is not implemented for Step Functions.
         
         Parameters
         ----------
-        identifier : str
-            Deployer specific identifier for the workflow to retrieve
-        metadata : str, optional, default None
-            Optional deployer specific metadata.
-        
         
         
         Returns
         -------
-        ArgoWorkflowsDeployedFlow
-            A `ArgoWorkflowsDeployedFlow` object representing the
-            deployed flow on argo workflows.
         """
         ...
     @classmethod
-    def list_argo_workflows(cls, flow_name: typing.Optional[str] = None):
+    def list_step_functions(cls, flow_name: typing.Optional[str] = None):
         ...
     @classmethod
-    def get_triggered_argo_workflows_run(cls, identifier: str, run_id: str, metadata: typing.Optional[str] = None):
+    def get_triggered_step_functions_run(cls, identifier: str, run_id: str, metadata: typing.Optional[str] = None):
         ...
     @classmethod
-    def from_step_functions(cls) -> "metaflow.plugins.aws.step_functions.step_functions_deployer_objects.StepFunctionsDeployedFlow":
+    def from_argo_workflows(cls, *, identifier: str, metadata: typing.Optional[str] = None) -> "metaflow.plugins.argo.argo_workflows_deployer_objects.ArgoWorkflowsDeployedFlow":
         """
-        This method is not currently implemented for Step Functions.
+        Retrieves a `ArgoWorkflowsDeployedFlow` object from an identifier and optional
+        metadata.
+        
         
-        Raises
-        ------
-        NotImplementedError
-            This method is not implemented for Step Functions.
         
         Parameters
         ----------
+        identifier : str
+            Deployer specific identifier for the workflow to retrieve
+        metadata : str, optional, default None
+            Optional deployer specific metadata.
+        
         
         
         Returns
         -------
+        ArgoWorkflowsDeployedFlow
+            A `ArgoWorkflowsDeployedFlow` object representing the
+            deployed flow on argo workflows.
         """
         ...
     @classmethod
-    def list_step_functions(cls, flow_name: typing.Optional[str] = None):
+    def list_argo_workflows(cls, flow_name: typing.Optional[str] = None):
         ...
     @classmethod
-    def get_triggered_step_functions_run(cls, identifier: str, run_id: str, metadata: typing.Optional[str] = None):
+    def get_triggered_argo_workflows_run(cls, identifier: str, run_id: str, metadata: typing.Optional[str] = None):
         ...
     ...
 

metaflow-stubs/runner/deployer_impl.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.848620                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.514732                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/runner/metaflow_runner.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.833948                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.495849                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,8 +10,8 @@ import typing
 import metaflow
 if typing.TYPE_CHECKING:
     import metaflow.client.core
-    import metaflow.runner.subprocess_manager
     import metaflow.runner.metaflow_runner
+    import metaflow.runner.subprocess_manager
 
 from ..client.core import Run as Run
 from ..plugins import get_runner_cli as get_runner_cli

metaflow-stubs/runner/nbdeploy.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.834959                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.496854                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/runner/nbrun.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.834556                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.496463                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/runner/subprocess_manager.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.847949                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.514022                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/runner/utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.846713                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.512747                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,8 +10,8 @@ import typing
 if typing.TYPE_CHECKING:
     import typing
     import types
-    import metaflow.runner.subprocess_manager
     import metaflow.runner.click_api
+    import metaflow.runner.subprocess_manager
 
 
 TYPE_CHECKING: bool

metaflow-stubs/system/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.835284                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.497208                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/system/system_logger.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.864848                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.531296                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/system/system_monitor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.864432                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.530877                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/tagging_util.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.813238                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.474608                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/tuple_util.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.820890                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.482468                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_configs/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.812684                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.474030                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_configs/config_options.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.841404                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.507327                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow._vendor.click.types
     import typing
+    import metaflow._vendor.click.types
 
 from .._vendor import click as click
 from .config_parameters import ConfigValue as ConfigValue

metaflow-stubs/user_configs/config_parameters.pyi

@@ -1,21 +1,21 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.817128                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.478618                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import abc
 import typing
-import metaflow
 import collections
+import abc
+import metaflow
 if typing.TYPE_CHECKING:
-    import abc
-    import metaflow.user_configs.config_parameters
     import typing
-    import collections.abc
+    import metaflow.user_configs.config_parameters
     import metaflow.parameters
+    import collections.abc
+    import abc
 
 from ..exception import MetaflowException as MetaflowException
 from ..parameters import Parameter as Parameter

metaflow-stubs/user_decorators/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.812916                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.474283                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_decorators/common.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.842855                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.508840                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_decorators/mutable_flow.pyi

@@ -1,19 +1,19 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.842373                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.508376                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.user_decorators.mutable_step
-    import metaflow.user_configs.config_parameters
     import typing
     import metaflow.flowspec
+    import metaflow.user_configs.config_parameters
     import functools
     import metaflow.parameters
+    import metaflow.user_decorators.mutable_step
 
 from ..exception import MetaflowException as MetaflowException
 from ..user_configs.config_parameters import ConfigValue as ConfigValue

metaflow-stubs/user_decorators/mutable_step.pyi

@@ -1,19 +1,19 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.843606                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.509623                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.user_decorator.mutable_flow
-    import metaflow.user_decorators.user_step_decorator
     import typing
     import metaflow.flowspec
-    import functools
     import metaflow.decorators
+    import metaflow.user_decorator.mutable_flow
+    import functools
+    import metaflow.user_decorators.user_step_decorator
 
 from ..exception import MetaflowException as MetaflowException
 from .user_step_decorator import StepMutator as StepMutator

metaflow-stubs/user_decorators/user_flow_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.819800                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.481351                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_decorators/user_step_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.818928                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.480482                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,12 +9,12 @@ from __future__ import annotations
 import typing
 import metaflow
 if typing.TYPE_CHECKING:
-    import metaflow.user_decorators.mutable_step
-    import metaflow.user_decorators.user_step_decorator
     import typing
     import metaflow.flowspec
-    import metaflow.datastore.inputs
     import metaflow.decorators
+    import metaflow.datastore.inputs
+    import metaflow.user_decorators.mutable_step
+    import metaflow.user_decorators.user_step_decorator
 
 from ..exception import MetaflowException as MetaflowException
 from ..user_configs.config_parameters import resolve_delayed_evaluator as resolve_delayed_evaluator
@@ -164,7 +164,7 @@ class UserStepDecorator(UserStepDecoratorBase, metaclass=UserStepDecoratorMeta):
             take an additional "inputs" argument).
         """
         ...
-    def post_step(self, step_name: str, flow: "metaflow.flowspec.FlowSpec", exception: typing.Optional[Exception] = None):
+    def post_step(self, step_name: str, flow: "metaflow.flowspec.FlowSpec", exception: typing.Optional[Exception] = None) -> typing.Union[Exception, None, typing.Tuple[typing.Optional[Exception], typing.Optional[typing.Dict[str, typing.Any]]]]:
         """
         Implement this method to perform any action after the execution of a step.
         
@@ -184,6 +184,19 @@ class UserStepDecorator(UserStepDecoratorBase, metaclass=UserStepDecoratorMeta):
             The flow object to which the step belongs.
         exception: Optional[Exception]
             The exception raised during the step execution, if any.
+        
+        Returns
+        -------
+        Optional[Union[Optional[Exception], Tuple[Optional[Exception], Optional[Dict[str, Any]]]]]
+            An exception (if None, the step is considered successful)
+            OR
+            A tuple containing:
+              - An exception to be raised (if None, the step is considered successful).
+              - A dictionary with values to pass to `self.next()`. If an empty dictionary
+                is returned, the default arguments to `self.next()` for this step will be
+                used. Return None if you do not want to call `self.next()` at all
+                (this is typically the case as the step will call it itself).
+        Note that returning None will gobble the exception.
         """
         ...
     @property

metaflow-stubs/version.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.17.1                                                                                 #
-# Generated on 2025-08-11T22:07:35.835073                                                            #
+# MF version: 2.17.2                                                                                 #
+# Generated on 2025-08-19T21:01:04.496967                                                            #
 ######################################################################################################
 
 from __future__ import annotations

{metaflow_stubs-2.17.1.dist-info → metaflow_stubs-2.17.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metaflow-stubs
-Version: 2.17.1
+Version: 2.17.2
 Summary: Metaflow Stubs: Stubs for the metaflow package
 Author: Metaflow Developers
 Author-email: help@metaflow.org
@@ -21,7 +21,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.7.0
 Description-Content-Type: text/markdown
-Requires-Dist: metaflow==2.17.1
+Requires-Dist: metaflow==2.17.2
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier