metaflow-stubs 2.12.28__py2.py3-none-any.whl → 2.12.30__py2.py3-none-any.whl

metaflow-stubs/{metadata → metadata_provider}/metadata.pyi

@@ -1,31 +1,19 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.498599                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.308183                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.metadata.metadata
-    import metaflow.exception
+    import metaflow.metadata_provider.metadata
 
-class MetaflowInternalError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-class MetaflowTaggingError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-def validate_tag(tag):
-    """
-    - Tag must be either of bytes-type or unicode-type.
-    - If tag is of bytes-type, it must be UTF-8 decodable
-    - If tag is of unicode-type, it must be UTF-8 encodable
-    - Tag may not be empty string.
-    - Tag cannot be too long (500 chars)
-    """
-    ...
+from ..exception import MetaflowInternalError as MetaflowInternalError
+from ..exception import MetaflowTaggingError as MetaflowTaggingError
+from ..tagging_util import validate_tag as validate_tag
 
 class DataArtifact(tuple, metaclass=type):
     """

metaflow-stubs/metadata_provider/util.pyi

@@ -0,0 +1,19 @@
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.308449                                                            #
+######################################################################################################
+
+from __future__ import annotations
+
+
+
+def copy_tree(src, dst, update = False):
+    ...
+
+def sync_local_metadata_to_datastore(metadata_local_dir, task_ds):
+    ...
+
+def sync_local_metadata_from_datastore(metadata_local_dir, task_ds):
+    ...
+

metaflow-stubs/metaflow_config.pyi

@@ -1,18 +1,13 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.437216                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.285281                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from .exception import MetaflowException as MetaflowException
 
 DATASTORE_LOCAL_DIR: str
 
@@ -40,6 +35,8 @@ DEFAULT_SECRETS_BACKEND_TYPE: None
 
 DEFAULT_SECRETS_ROLE: None
 
+DEFAULT_FROM_DEPLOYMENT_IMPL: str
+
 USER: None
 
 DATASTORE_SYSROOT_LOCAL: None

metaflow-stubs/metaflow_current.pyi

@@ -1,19 +1,20 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.563771                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.392605                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.cards.component_serializer
+    import metaflow
     import metaflow.events
+    import metaflow.plugins.cards.component_serializer
     import typing
-    import metaflow
     import metaflow.metaflow_current
 
+
 TYPE_CHECKING: bool
 
 TEMPDIR: str
@@ -221,7 +222,25 @@ class Current(object, metaclass=type):
         """
         ...
     @property
-    def graph(_):
+    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":
@@ -253,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
@@ -337,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/mflog/__init__.pyi

@@ -0,0 +1,6 @@
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.329509                                                            #
+######################################################################################################
+

metaflow-stubs/mflog/mflog.pyi

@@ -1,8 +1,8 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.498980                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.329566                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
@@ -10,13 +10,60 @@ import typing
 if typing.TYPE_CHECKING:
     import datetime
 
+
 VERSION: bytes
 
 RE: bytes
 
+class MFLogline(tuple, metaclass=type):
+    """
+    MFLogline(should_persist, version, utc_tstamp_str, logsource, id, msg, utc_tstamp)
+    """
+    @staticmethod
+    def __new__(_cls, should_persist, version, utc_tstamp_str, logsource, id, msg, utc_tstamp):
+        """
+        Create new instance of MFLogline(should_persist, version, utc_tstamp_str, logsource, id, msg, utc_tstamp)
+        """
+        ...
+    def __repr__(self):
+        """
+        Return a nicely formatted representation string
+        """
+        ...
+    def __getnewargs__(self):
+        """
+        Return self as a plain tuple.  Used by copy and pickle.
+        """
+        ...
+    ...
+
 ISOFORMAT: str
 
 MISSING_TIMESTAMP: datetime.datetime
 
 MISSING_TIMESTAMP_STR: str
 
+def utc_to_local(x):
+    ...
+
+def decorate(source, line, version = b'0', now = None, lineid = None):
+    ...
+
+def is_structured(line):
+    ...
+
+def parse(line):
+    ...
+
+def set_should_persist(line):
+    ...
+
+def unset_should_persist(line):
+    ...
+
+def refine(line, prefix = None, suffix = None):
+    ...
+
+def merge_logs(logs):
+    ...
+

metaflow-stubs/multicore_utils.pyi

@@ -1,8 +1,8 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.437707                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.285795                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
@@ -10,6 +10,7 @@ import typing
 if typing.TYPE_CHECKING:
     import typing
 
+
 class MulticoreException(Exception, metaclass=type):
     ...
 

metaflow-stubs/parameters.pyi

@@ -1,34 +1,22 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.439142                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.287252                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.parameters
     import metaflow._vendor.click.types
     import typing
-    import metaflow.exception
 
-class ParameterFieldFailed(metaflow.exception.MetaflowException, metaclass=type):
-    def __init__(self, name, field):
-        ...
-    ...
-
-class ParameterFieldTypeMismatch(metaflow.exception.MetaflowException, metaclass=type):
-    def __init__(self, msg):
-        ...
-    ...
-
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from ._vendor import click as click
+from .exception import ParameterFieldFailed as ParameterFieldFailed
+from .exception import ParameterFieldTypeMismatch as ParameterFieldTypeMismatch
+from .exception import MetaflowException as MetaflowException
 
 class ParameterContext(tuple, metaclass=type):
     """
@@ -184,3 +172,5 @@ class Parameter(object, metaclass=type):
 def add_custom_parameters(deploy_mode = False):
     ...
 
+JSONType: JSONTypeClass
+

metaflow-stubs/plugins/__init__.pyi

@@ -1,38 +1,55 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.453555                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.291226                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.unbounded_foreach
-    import metaflow.plugins.cards.card_modules.card
 
-CLIS_DESC: list
+from . import test_unbounded_foreach_decorator as test_unbounded_foreach_decorator
+from .test_unbounded_foreach_decorator import InternalTestUnboundedForeachInput as InternalTestUnboundedForeachInput
+from . import catch_decorator as catch_decorator
+from . import retry_decorator as retry_decorator
+from . import cards as cards
+from . import aws as aws
+from . import secrets as secrets
+from . import airflow as airflow
+from . import kubernetes as kubernetes
+from . import resources_decorator as resources_decorator
+from . import timeout_decorator as timeout_decorator
+from . import frameworks as frameworks
+from . import parallel_decorator as parallel_decorator
+from . import pypi as pypi
+from . import environment_decorator as environment_decorator
+from . import datatools as datatools
+from . import argo as argo
+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
+from . import debug_logger as debug_logger
+from . import debug_monitor as debug_monitor
+from .cards.card_modules.basic import BlankCard as BlankCard
+from .cards.card_modules.basic import DefaultCard as DefaultCard
+from .cards.card_modules.basic import DefaultCardJSON as DefaultCardJSON
+from .cards.card_modules.basic import ErrorCard as ErrorCard
+from .cards.card_modules.basic import TaskSpecCard as TaskSpecCard
+from .cards.card_modules.test_cards import TestEditableCard as TestEditableCard
+from .cards.card_modules.test_cards import TestEditableCard2 as TestEditableCard2
+from .cards.card_modules.test_cards import TestErrorCard as TestErrorCard
+from .cards.card_modules.test_cards import TestMockCard as TestMockCard
+from .cards.card_modules.test_cards import TestNonEditableCard as TestNonEditableCard
+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 package_cli as package_cli
+from . import tag_cli as tag_cli
 
-class InternalTestUnboundedForeachInput(metaflow.unbounded_foreach.UnboundedForeachInput, metaclass=type):
-    """
-    Test class that wraps around values (any iterator) and simulates an
-    unbounded-foreach instead of a bounded foreach.
-    """
-    def __init__(self, iterable):
-        ...
-    def __iter__(self):
-        ...
-    def __next__(self):
-        ...
-    def __getitem__(self, key):
-        ...
-    def __len__(self):
-        ...
-    def __str__(self):
-        ...
-    def __repr__(self):
-        ...
-    ...
+CLIS_DESC: list
 
 STEP_DECORATORS_DESC: list
 
@@ -44,6 +61,8 @@ METADATA_PROVIDERS_DESC: list
 
 DATASTORES_DESC: list
 
+DATACLIENTS_DESC: list
+
 SIDECARS_DESC: list
 
 LOGGING_SIDECARS_DESC: list
@@ -75,6 +94,8 @@ METADATA_PROVIDERS: list
 
 DATASTORES: list
 
+DATACLIENTS: list
+
 SIDECARS: dict
 
 LOGGING_SIDECARS: dict
@@ -93,138 +114,5 @@ DEPLOYER_IMPL_PROVIDERS: list
 
 MF_EXTERNAL_CARDS: list
 
-class BlankCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {"title": ""}, components = [], graph = None):
-        ...
-    def render(self, task, components = [], runtime = False):
-        ...
-    def render_runtime(self, task, data):
-        ...
-    def refresh(self, task, data):
-        ...
-    def reload_content_token(self, task, data):
-        """
-        The reload token will change when the component array has changed in the Metaflow card.
-        The change in the component array is signified by the change in the component_update_ts.
-        """
-        ...
-    ...
-
-class DefaultCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {"only_repr": True}, components = [], graph = None):
-        ...
-    def render(self, task, runtime = False):
-        ...
-    def render_runtime(self, task, data):
-        ...
-    def refresh(self, task, data):
-        ...
-    def reload_content_token(self, task, data):
-        """
-        The reload token will change when the component array has changed in the Metaflow card.
-        The change in the component array is signified by the change in the component_update_ts.
-        """
-        ...
-    ...
-
-class DefaultCardJSON(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {"only_repr": True}, components = [], graph = None):
-        ...
-    def render(self, task):
-        ...
-    ...
-
-class ErrorCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {}, components = [], graph = None):
-        ...
-    def reload_content_token(self, task, data):
-        """
-        The reload token will change when the component array has changed in the Metaflow card.
-        The change in the component array is signified by the change in the component_update_ts.
-        """
-        ...
-    def render(self, task, stack_trace = None):
-        ...
-    ...
-
-class TaskSpecCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def render(self, task):
-        ...
-    ...
-
-class TestEditableCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {}, components = [], graph = None):
-        ...
-    def render(self, task):
-        ...
-    ...
-
-class TestEditableCard2(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {}, components = [], graph = None):
-        ...
-    def render(self, task):
-        ...
-    ...
-
-class TestErrorCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def render(self, task):
-        ...
-    ...
-
-class TestMockCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {"key": "dummy_key"}, **kwargs):
-        ...
-    def render(self, task):
-        ...
-    ...
-
-class TestNonEditableCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {}, components = [], graph = None):
-        ...
-    def render(self, task):
-        ...
-    ...
-
-class TestPathSpecCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def render(self, task):
-        ...
-    ...
-
-class TestTimeoutCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    def __init__(self, options = {"timeout": 50}, **kwargs):
-        ...
-    def render(self, task):
-        ...
-    ...
-
-class TestRefreshCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    """
-    This card takes no components and helps test the `current.card.refresh(data)` interface.
-    """
-    def render(self, task) -> str:
-        ...
-    def render_runtime(self, task, data):
-        ...
-    def refresh(self, task, data):
-        ...
-    ...
-
-class TestRefreshComponentCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):
-    """
-    This card takes components and helps test the `current.card.components["A"].update()`
-    interface
-    """
-    def __init__(self, options = {}, components = [], graph = None):
-        ...
-    def render(self, task) -> str:
-        ...
-    def render_runtime(self, task, data):
-        ...
-    def refresh(self, task, data):
-        ...
-    def reload_content_token(self, task, data):
-        ...
-    ...
-
 CARDS: list
 

metaflow-stubs/plugins/airflow/__init__.pyi

@@ -1,9 +1,16 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.479224                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30                                                                                #
+# Generated on 2024-11-13T13:50:31.313978                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
+from . import airflow_utils as airflow_utils
+from . import airflow_decorator as airflow_decorator
+from . import exception as exception
+from . import sensors as sensors
+from . import airflow as airflow
+from . import airflow_cli as airflow_cli
+