metaflow-stubs 2.11.15__py2.py3-none-any.whl → 2.12.0__py2.py3-none-any.whl

metaflow-stubs/cards.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.340953                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.178939                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -10,10 +10,10 @@ import typing
 if typing.TYPE_CHECKING:
     import metaflow
     import metaflow.plugins.cards.card_modules.card
-    import metaflow.plugins.cards.card_modules.basic
     import metaflow.plugins.cards.card_modules.components
-    import metaflow.plugins.cards.card_client
     import typing
+    import metaflow.plugins.cards.card_client
+    import metaflow.plugins.cards.card_modules.basic
 
 def get_cards(task: typing.Union[str, "metaflow.Task"], id: typing.Optional[str] = None, type: typing.Optional[str] = None, follow_resumed: bool = True) -> metaflow.plugins.cards.card_client.CardContainer:
     """

metaflow-stubs/cli.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.347588                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.188718                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.metaflow_current
+    import metaflow.exception
 
 def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:
     """
@@ -31,16 +31,23 @@ def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:
     """
     ...
 
-current: metaflow.metaflow_current.Current
-
-def validate_tags(tags, existing_tags = None):
+def get_metadata() -> str:
     """
-    Raises MetaflowTaggingError if invalid based on these rules:
+    Returns the current Metadata provider.
     
-    Tag set size is too large. But it's OK if tag set is not larger
-    than an existing tag set (if provided).
+    If this is not set explicitly using `metadata`, the default value is
+    determined through the Metaflow configuration. You can use this call to
+    check that your configuration is set up properly.
     
-    Then, we validate each tag.  See validate_tag()
+    If multiple configuration profiles are present, this call returns the one
+    selected through the `METAFLOW_PROFILE` environment variable.
+    
+    Returns
+    -------
+    str
+        Information about the Metadata provider currently selected. This information typically
+        returns provider specific information (like URL for remote providers or local paths for
+        local providers).
     """
     ...
 
@@ -54,16 +61,6 @@ class MetaflowException(Exception, metaclass=type):
         ...
     ...
 
-DATASTORES: list
-
-ENVIRONMENTS: list
-
-LOGGING_SIDECARS: dict
-
-METADATA_PROVIDERS: list
-
-MONITOR_SIDECARS: dict
-
 DEFAULT_DATASTORE: str
 
 DEFAULT_ENVIRONMENT: str
@@ -76,6 +73,20 @@ DEFAULT_MONITOR: str
 
 DEFAULT_PACKAGE_SUFFIXES: str
 
+current: metaflow.metaflow_current.Current
+
+LOG_SOURCES: list
+
+DATASTORES: list
+
+ENVIRONMENTS: list
+
+LOGGING_SIDECARS: dict
+
+METADATA_PROVIDERS: list
+
+MONITOR_SIDECARS: dict
+
 class PyLint(object, metaclass=type):
     def __init__(self, fname):
         ...
@@ -85,7 +96,16 @@ class PyLint(object, metaclass=type):
         ...
     ...
 
-LOG_SOURCES: list
+def validate_tags(tags, existing_tags = None):
+    """
+    Raises MetaflowTaggingError if invalid based on these rules:
+    
+    Tag set size is too large. But it's OK if tag set is not larger
+    than an existing tag set (if provided).
+    
+    Then, we validate each tag.  See validate_tag()
+    """
+    ...
 
 UBF_CONTROL: str
 
@@ -115,7 +135,7 @@ def logger(body = "", system_msg = False, head = "", bad = False, timestamp = Tr
 def common_run_options(func):
     ...
 
-def write_run_id(run_id_file, run_id):
+def write_file(file_path, content):
     ...
 
 def before_run(obj, tags, decospecs):

metaflow-stubs/client/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.344802                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.181932                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.client.core
-    import datetime
     import metaflow.events
+    import datetime
     import typing
 
 def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:

metaflow-stubs/client/core.pyi

@@ -1,21 +1,21 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.326629                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.165323                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.events
-    import datetime
     import metaflow
+    import datetime
     import metaflow.client.core
-    import metaflow.exception
-    import tarfile
-    import metaflow.metaflow_current
     import typing
+    import metaflow.metaflow_current
+    import metaflow.events
+    import tarfile
+    import metaflow.exception
 
 current: metaflow.metaflow_current.Current
 

metaflow-stubs/client/filecache.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.348707                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.189755                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.datastore.content_addressed_store
+    import metaflow.exception
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/clone_util.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.345294                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.187329                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/events.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.329088                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.167869                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/exception.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.316765                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.156081                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/flowspec.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.328360                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.166982                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,10 +9,10 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.parameters
-    import metaflow.exception
     import metaflow.datastore.inputs
     import metaflow.unbounded_foreach
     import typing
+    import metaflow.exception
 
 class DelayedEvaluationParameter(object, metaclass=type):
     def __init__(self, name, field, fun):

metaflow-stubs/generated_for.txt

@@ -1 +1 @@
-2.11.15 2024-05-08T22:46:02.299859
+2.12.0 2024-05-28T09:55:27.139119

metaflow-stubs/includefile.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.338736                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.176873                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,10 +9,10 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.parameters
-    import io
-    import metaflow.plugins.datatools.s3.s3
     import metaflow._vendor.click.types
+    import metaflow.plugins.datatools.s3.s3
     import typing
+    import io
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/metadata/metadata.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.370956                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.211687                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/metadata/util.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.425132                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.263111                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/metaflow_config.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.318108                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.157447                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -28,6 +28,8 @@ DEFAULT_PACKAGE_SUFFIXES: str
 
 DEFAULT_AWS_CLIENT_PROVIDER: str
 
+DEFAULT_GCP_CLIENT_PROVIDER: str
+
 DEFAULT_SECRETS_BACKEND_TYPE: None
 
 DEFAULT_SECRETS_ROLE: None
@@ -84,6 +86,10 @@ DATATOOLS_LOCALROOT: None
 
 AWS_SECRETS_MANAGER_DEFAULT_REGION: None
 
+GCP_SECRET_MANAGER_PREFIX: None
+
+AZURE_KEY_VAULT_PREFIX: None
+
 ARTIFACT_LOCALROOT: str
 
 CARD_SUFFIX: str
@@ -126,6 +132,8 @@ INCLUDE_FOREACH_STACK: bool
 
 MAXIMUM_FOREACH_VALUE_CHARS: int
 
+DEFAULT_RUNTIME_LIMIT: int
+
 UI_URL: None
 
 CONTACT_INFO: dict
@@ -194,6 +202,10 @@ ARGO_WORKFLOWS_KUBERNETES_SECRETS: str
 
 ARGO_WORKFLOWS_ENV_VARS_TO_SKIP: str
 
+KUBERNETES_JOBSET_GROUP: str
+
+KUBERNETES_JOBSET_VERSION: str
+
 ARGO_EVENTS_SERVICE_ACCOUNT: None
 
 ARGO_EVENTS_EVENT_BUS: str

metaflow-stubs/metaflow_current.pyi

@@ -1,18 +1,18 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.427940                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.268101                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.events
+    import metaflow.plugins.cards.component_serializer
     import metaflow
-    import metaflow.metaflow_current
     import typing
-    import metaflow.plugins.cards.component_serializer
+    import metaflow.metaflow_current
+    import metaflow.events
 
 TYPE_CHECKING: bool
 

metaflow-stubs/mflog/mflog.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.371358                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.212074                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/multicore_utils.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.318632                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.157949                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/parameters.pyi

@@ -1,17 +1,17 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.320033                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.159363                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.parameters
     import typing
     import metaflow._vendor.click.types
+    import metaflow.exception
 
 class ParameterFieldFailed(metaflow.exception.MetaflowException, metaclass=type):
     def __init__(self, name, field):

metaflow-stubs/plugins/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.331037                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.169745                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -52,6 +52,10 @@ SENSOR_FLOW_DECORATORS: list
 
 SECRETS_PROVIDERS_DESC: list
 
+GCP_CLIENT_PROVIDERS_DESC: list
+
+AZURE_CLIENT_PROVIDERS_DESC: list
+
 def get_plugin_cli():
     ...
 
@@ -75,6 +79,10 @@ AWS_CLIENT_PROVIDERS: list
 
 SECRETS_PROVIDERS: list
 
+AZURE_CLIENT_PROVIDERS: list
+
+GCP_CLIENT_PROVIDERS: list
+
 MF_EXTERNAL_CARDS: list
 
 class BlankCard(metaflow.plugins.cards.card_modules.card.MetaflowCard, metaclass=type):

metaflow-stubs/plugins/airflow/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.351727                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.194893                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/airflow.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.379351                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.244542                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
-    import metaflow.metaflow_current
     import metaflow._vendor.click.types
+    import metaflow.metaflow_current
 
 current: metaflow.metaflow_current.Current
 
@@ -42,6 +42,8 @@ AIRFLOW_KUBERNETES_STARTUP_TIMEOUT_SECONDS: int
 
 AWS_SECRETS_MANAGER_DEFAULT_REGION: None
 
+GCP_SECRET_MANAGER_PREFIX: None
+
 AZURE_STORAGE_BLOB_SERVICE_ENDPOINT: None
 
 CARD_AZUREROOT: None
@@ -70,6 +72,8 @@ SERVICE_HEADERS: dict
 
 SERVICE_INTERNAL_URL: None
 
+AZURE_KEY_VAULT_PREFIX: None
+
 class DelayedEvaluationParameter(object, metaclass=type):
     def __init__(self, name, field, fun):
         ...
@@ -94,7 +98,11 @@ class Kubernetes(object, metaclass=type):
         ...
     def launch_job(self, **kwargs):
         ...
-    def create_job(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, step_cli, docker_image, docker_image_pull_policy, service_account = None, secrets = None, node_selector = None, namespace = None, cpu = None, gpu = None, gpu_vendor = None, disk = None, memory = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, run_time_limit = None, env = None, persistent_volume_claims = None, tolerations = None, labels = None, shared_memory = None, port = None):
+    def create_jobset(self, job_spec = None, run_id = None, step_name = None, task_id = None, namespace = None, env = None, num_parallel = None, port = None, annotations = None, labels = None):
+        ...
+    def create_job_object(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, step_cli, docker_image, docker_image_pull_policy, service_account = None, secrets = None, node_selector = None, namespace = None, cpu = None, gpu = None, gpu_vendor = None, disk = None, memory = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, run_time_limit = None, env = None, persistent_volume_claims = None, tolerations = None, labels = None, shared_memory = None, port = None, name_pattern = None, num_parallel = None):
+        ...
+    def create_k8sjob(self, job):
         ...
     def wait(self, stdout_location, stderr_location, echo = None):
         ...

metaflow-stubs/plugins/airflow/airflow_cli.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.380594                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.245677                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
-    import metaflow.metaflow_current
     import metaflow.decorators
+    import metaflow.metaflow_current
 
 current: metaflow.metaflow_current.Current
 

metaflow-stubs/plugins/airflow/airflow_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.376978                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.242239                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/airflow_utils.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.376506                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.241793                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/exception.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.377248                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.242497                                        #
 ##################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.377518                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.242766                                        #
 ##################################################################################
 
 from __future__ import annotations

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

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.423481                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.264525                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.decorators
+    import metaflow.exception
 
 class AirflowException(metaflow.exception.MetaflowException, metaclass=type):
     def __init__(self, msg):

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

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.423962                                        #
+# MF version: 2.12.0                                                             #
+# Generated on 2024-05-28T09:55:27.265007                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.decorators
     import metaflow.plugins.airflow.sensors.base_sensor
+    import metaflow.exception
 
 class AirflowSensorDecorator(metaflow.decorators.FlowDecorator, metaclass=type):
     def __init__(self, *args, **kwargs):