ob-metaflow-stubs 6.0.3.125__py2.py3-none-any.whl → 6.0.3.127__py2.py3-none-any.whl

metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi

@@ -1,6 +1,6 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.098985                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.918373                                                            #
 ######################################################################################################
 

metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.099032                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.918420                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/multicore_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.093965                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.910089                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/parameters.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.095124                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.911311                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,22 +10,24 @@ import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow._vendor.click.types
-    import metaflow.parameters
     import typing
+    import metaflow.parameters
 
 from ._vendor import click as click
 from .exception import ParameterFieldFailed as ParameterFieldFailed
 from .exception import ParameterFieldTypeMismatch as ParameterFieldTypeMismatch
 from .exception import MetaflowException as MetaflowException
 
+TYPE_CHECKING: bool
+
 class ParameterContext(tuple, metaclass=type):
     """
-    ParameterContext(flow_name, user_name, parameter_name, logger, ds_type)
+    ParameterContext(flow_name, user_name, parameter_name, logger, ds_type, configs)
     """
     @staticmethod
-    def __new__(_cls, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str):
+    def __new__(_cls, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str, configs: typing.Optional["ConfigValue"]):
         """
-        Create new instance of ParameterContext(flow_name, user_name, parameter_name, logger, ds_type)
+        Create new instance of ParameterContext(flow_name, user_name, parameter_name, logger, ds_type, configs)
         """
         ...
     def __repr__(self):
@@ -38,7 +40,7 @@ class ParameterContext(tuple, metaclass=type):
         Return self as a plain tuple.  Used by copy and pickle.
         """
         ...
-    def __init__(self, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str):
+    def __init__(self, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str, configs: typing.Optional["ConfigValue"]):
         ...
     ...
 
@@ -52,6 +54,13 @@ def flow_context(flow_cls):
 
 context_proto: None
 
+def replace_flow_context(flow_cls):
+    """
+    Replace the current flow context with a new flow class. This is used
+    when we change the current flow class after having run user configuration functions
+    """
+    ...
+
 class JSONTypeClass(metaflow._vendor.click.types.ParamType, metaclass=type):
     def convert(self, value, param, ctx):
         ...
@@ -86,7 +95,7 @@ class DeployTimeField(object, metaclass=type):
 def deploy_time_eval(value):
     ...
 
-def set_parameter_context(flow_name, echo, datastore):
+def set_parameter_context(flow_name, echo, datastore, configs):
     ...
 
 class DelayedEvaluationParameter(object, metaclass=type):
@@ -135,7 +144,11 @@ class Parameter(object, metaclass=type):
     ----------
     name : str
         User-visible parameter name.
-    default : str or float or int or bool or `JSONType` or a function.
+    default : Union[str, float, int, bool, Dict[str, Any],
+                Callable[
+                    [ParameterContext], Union[str, float, int, bool, Dict[str, Any]]
+                ],
+            ], optional, default None
         Default value for the parameter. Use a special `JSONType` class to
         indicate that the value must be a valid JSON object. A function
         implies that the parameter corresponds to a *deploy-time parameter*.
@@ -144,15 +157,19 @@ class Parameter(object, metaclass=type):
         If `default` is not specified, define the parameter type. Specify
         one of `str`, `float`, `int`, `bool`, or `JSONType`. If None, defaults
         to the type of `default` or `str` if none specified.
-    help : str, optional
+    help : str, optional, default None
         Help text to show in `run --help`.
-    required : bool, default False
-        Require that the user specified a value for the parameter.
-        `required=True` implies that the `default` is not used.
-    show_default : bool, default True
-        If True, show the default value in the help text.
+    required : bool, optional, default None
+        Require that the user specified a value for the parameter. Note that if
+        a default is provide, the required flag is ignored.
+        A value of None is equivalent to False.
+    show_default : bool, optional, default None
+        If True, show the default value in the help text. A value of None is equivalent
+        to True.
     """
-    def __init__(self, name: str, default: typing.Union[str, float, int, bool, typing.Dict[str, typing.Any], typing.Callable[[], typing.Union[str, float, int, bool, typing.Dict[str, typing.Any]]], None] = None, type: typing.Union[typing.Type[str], typing.Type[float], typing.Type[int], typing.Type[bool], JSONTypeClass, None] = None, help: typing.Optional[str] = None, required: bool = False, show_default: bool = True, **kwargs: typing.Dict[str, typing.Any]):
+    def __init__(self, name: str, default: typing.Union[str, float, int, bool, typing.Dict[str, typing.Any], typing.Callable[[ParameterContext], typing.Union[str, float, int, bool, typing.Dict[str, typing.Any]]], None] = None, type: typing.Union[typing.Type[str], typing.Type[float], typing.Type[int], typing.Type[bool], JSONTypeClass, None] = None, help: typing.Optional[str] = None, required: typing.Optional[bool] = None, show_default: typing.Optional[bool] = None, **kwargs: typing.Dict[str, typing.Any]):
+        ...
+    def init(self):
         ...
     def __repr__(self):
         ...

metaflow-stubs/plugins/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.098502                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.917863                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,22 +9,22 @@ 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 frameworks as frameworks
-from . import parallel_decorator as parallel_decorator
-from . import environment_decorator as environment_decorator
-from . import resources_decorator as resources_decorator
-from . import catch_decorator as catch_decorator
+from . import cards as cards
+from . import aws as aws
 from . import datatools as datatools
 from . import gcp as gcp
 from . import storage_executor as storage_executor
-from . import cards as cards
+from . import parallel_decorator as parallel_decorator
+from . import retry_decorator as retry_decorator
+from . import secrets as secrets
+from . import catch_decorator as catch_decorator
+from . import resources_decorator as resources_decorator
+from . import argo as argo
 from . import pypi as pypi
-from . import aws as aws
+from . import environment_decorator as environment_decorator
+from . import frameworks as frameworks
 from . import timeout_decorator as timeout_decorator
-from . import argo as argo
 from . import kubernetes as kubernetes
-from . import secrets as secrets
-from . import retry_decorator as retry_decorator
 from . import airflow as airflow
 from . import events_decorator as events_decorator
 from . import project_decorator as project_decorator
@@ -45,9 +45,6 @@ from .cards.card_modules.test_cards import TestPathSpecCard as TestPathSpecCard
 from .cards.card_modules.test_cards import TestTimeoutCard as TestTimeoutCard
 from .cards.card_modules.test_cards import TestRefreshCard as TestRefreshCard
 from .cards.card_modules.test_cards import TestRefreshComponentCard as TestRefreshComponentCard
-from . import logs_cli as logs_cli
-from . import tag_cli as tag_cli
-from . import package_cli as package_cli
 from . import snowflake as snowflake
 from . import perimeters as perimeters
 
@@ -86,6 +83,9 @@ DEPLOYER_IMPL_PROVIDERS_DESC: list
 def get_plugin_cli():
     ...
 
+def get_plugin_cli_path():
+    ...
+
 STEP_DECORATORS: list
 
 FLOW_DECORATORS: list

metaflow-stubs/plugins/airflow/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.128439                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.946680                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/airflow_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.160922                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.970852                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/exception.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.160043                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.969743                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.161171                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.971114                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.172529                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.011690                                                            #
 ######################################################################################################
 
 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.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.172792                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.011966                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.173019                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.012196                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.127453                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.945171                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,7 +10,4 @@ from __future__ import annotations
 from . import argo_events as argo_events
 from . import argo_workflows_decorator as argo_workflows_decorator
 from . import argo_workflows_deployer as argo_workflows_deployer
-from . import argo_client as argo_client
-from . import argo_workflows as argo_workflows
-from . import argo_workflows_cli as argo_workflows_cli
 

metaflow-stubs/plugins/argo/argo_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.150711                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.024942                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_events.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.149428                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.964612                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_workflows.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.154729                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.029241                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -22,6 +22,7 @@ from ..kubernetes.kube_utils import qos_requests_and_limits as qos_requests_and_
 from ..kubernetes.kubernetes import parse_kube_keyvalue_list as parse_kube_keyvalue_list
 from ..kubernetes.kubernetes import validate_kube_labels as validate_kube_labels
 from ..kubernetes.kubernetes_jobsets import KubernetesArgoJobSet as KubernetesArgoJobSet
+from ...user_configs.config_options import ConfigInput as ConfigInput
 from .argo_client import ArgoClient as ArgoClient
 
 inf: float

metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.149734                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.964919                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.flowspec
     import metaflow.decorators
+    import metaflow.flowspec
     import metaflow.graph
 
 from ...metaflow_current import current as current

metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.150135                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.965320                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.172161                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.011296                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -77,6 +77,34 @@ class ArgoWorkflowsTriggeredRun(metaflow.runner.deployer.TriggeredRun, metaclass
             True if the command was successful, False otherwise.
         """
         ...
+    def wait_for_completion(self, timeout: typing.Optional[int] = None):
+        """
+        Wait for the workflow to complete or timeout.
+        
+        Parameters
+        ----------
+        timeout : int, optional, default None
+            Maximum time in seconds to wait for workflow completion.
+            If None, waits indefinitely.
+        
+        Raises
+        ------
+        TimeoutError
+            If the workflow does not complete within the specified timeout period.
+        """
+        ...
+    @property
+    def is_running(self):
+        """
+        Check if the workflow is currently running.
+        
+        Returns
+        -------
+        bool
+            True if the workflow status is either 'Pending' or 'Running',
+            False otherwise.
+        """
+        ...
     @property
     def status(self) -> typing.Optional[str]:
         """

metaflow-stubs/plugins/aws/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.126832                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.942342                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/aws_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.148775                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.958164                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/aws_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.148011                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.957388                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.148479                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.957831                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/batch.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.168721                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.978016                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/batch_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.167952                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.977220                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/batch_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.169286                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.978585                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -102,7 +102,7 @@ class BatchDecorator(metaflow.decorators.StepDecorator, metaclass=type):
         depend on the `log driver` chosen. Validation of these options is not supported yet.
         Example: [`awslogs-group:aws/batch/job`]
     """
-    def __init__(self, attributes = None, statically_defined = False):
+    def init(self):
         ...
     def step_init(self, flow, graph, step, decos, environment, flow_datastore, logger):
         ...

metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.148979                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.958368                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.169586                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.978897                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,9 +10,9 @@ import abc
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
+    import metaflow.plugins.secrets
     import metaflow.exception
     import abc
-    import metaflow.plugins.secrets
 
 from ....exception import MetaflowException as MetaflowException
 from ...secrets import SecretsProvider as SecretsProvider

metaflow-stubs/plugins/aws/step_functions/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.148233                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.957600                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,5 +9,4 @@ from __future__ import annotations
 
 from . import schedule_decorator as schedule_decorator
 from . import step_functions_deployer as step_functions_deployer
-from . import production_token as production_token
 

metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.187552                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.033516                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.166083                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:50.975500                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/step_functions/step_functions.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.187253                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.031677                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -13,6 +13,7 @@ if typing.TYPE_CHECKING:
 
 from ....exception import MetaflowException as MetaflowException
 from ....parameters import deploy_time_eval as deploy_time_eval
+from ....user_configs.config_options import ConfigInput as ConfigInput
 from ..batch.batch import Batch as Batch
 from .event_bridge_client import EventBridgeClient as EventBridgeClient
 from .step_functions_client import StepFunctionsClient as StepFunctionsClient

metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1)                                                   #
-# Generated on 2024-12-10T22:06:22.187878                                                            #
+# MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-12-17T21:56:51.033889                                                            #
 ######################################################################################################
 
 from __future__ import annotations