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

metaflow-stubs/plugins/cards/card_modules/chevron/main.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.426572                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.097882                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_modules/chevron/metadata.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.426233                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.097558                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_modules/chevron/renderer.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.426114                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.097438                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_modules/chevron/tokenizer.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.425648                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.096878                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_modules/components.pyi

@@ -1,17 +1,17 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.369459                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.040139                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import typing
+    import metaflow.plugins.cards.card_modules.components
     import metaflow.plugins.cards.card_modules.card
+    import typing
     import metaflow.plugins.cards.card_modules.basic
-    import metaflow.plugins.cards.card_modules.components
 
 class LogComponent(metaflow.plugins.cards.card_modules.basic.DefaultComponent, metaclass=type):
     def __init__(self, data = None):

metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.410784                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.082681                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.411014                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.082903                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_modules/test_cards.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.360493                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.031662                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.cards.card_modules.card
     import metaflow
+    import metaflow.plugins.cards.card_modules.card
 
 class MetaflowCard(object, metaclass=type):
     def __init__(self, options = {}, components = [], graph = None):

metaflow-stubs/plugins/cards/card_resolver.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.391235                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.047660                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/component_serializer.pyi

@@ -1,17 +1,17 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.388446                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.044951                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
+    import metaflow.plugins.cards.card_modules.components
     import metaflow.plugins.cards.card_modules.card
+    import metaflow.exception
     import metaflow.plugins.cards.card_modules.basic
-    import metaflow.plugins.cards.card_modules.components
 
 class MetaflowCardComponent(object, metaclass=type):
     @property

metaflow-stubs/plugins/cards/exception.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.386734                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.043369                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/catch_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.353457                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.024658                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.351574                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.023303                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/local.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.363880                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.035027                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/s3/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.375169                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.053859                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/s3/s3.pyi

@@ -1,19 +1,19 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.335225                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.007228                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import io
+    import metaflow.plugins.datatools.s3.s3
     import metaflow.exception
     import metaflow.datastore.inputs
-    import metaflow.plugins.datatools.s3.s3
-    import metaflow.metaflow_current
     import typing
+    import io
+    import metaflow.metaflow_current
 
 TYPE_CHECKING: bool
 
@@ -250,6 +250,9 @@ S3_SERVER_SIDE_ENCRYPTION: None
 
 TEMPDIR: str
 
+def namedtuple_with_defaults(typename, field_descr, defaults = ()):
+    ...
+
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):
         ...

metaflow-stubs/plugins/datatools/s3/s3tail.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.422879                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.094629                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/s3/s3util.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.363224                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.034366                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_logger.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.356183                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.027572                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_monitor.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.356483                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.027846                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/environment_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.354040                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.024055                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/events_decorator.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.354862                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.026176                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.metaflow_current
+    import metaflow.decorators
 
 current: metaflow.metaflow_current.Current
 

metaflow-stubs/plugins/frameworks/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.352815                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.025794                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/frameworks/pytorch.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.397544                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.078817                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.plugins.parallel_decorator
-    import metaflow.decorators
     import metaflow.metaflow_current
+    import metaflow.decorators
 
 current: metaflow.metaflow_current.Current
 

metaflow-stubs/plugins/gcp/__init__.pyi

@@ -1,9 +1,12 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.355391                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.027305                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 
+def get_credentials(scopes, *args, **kwargs):
+    ...
+

metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi

@@ -0,0 +1,73 @@
+##################################################################################
+#                       Auto-generated Metaflow stub file                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.082220                                        #
+##################################################################################
+
+from __future__ import annotations
+
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.plugins.secrets
+    import abc
+    import metaflow.exception
+
+class MetaflowException(Exception, metaclass=type):
+    def __init__(self, msg = "", lineno = None):
+        ...
+    def __str__(self):
+        ...
+    ...
+
+class SecretsProvider(abc.ABC, metaclass=abc.ABCMeta):
+    def get_secret_as_dict(self, secret_id, options = {}, role = None) -> typing.Dict[str, str]:
+        """
+        Retrieve the secret from secrets backend, and return a dictionary of
+        environment variables.
+        """
+        ...
+    ...
+
+def get_credentials(scopes, *args, **kwargs):
+    ...
+
+GCP_SECRET_MANAGER_PREFIX: None
+
+class MetaflowGcpSecretsManagerBadResponse(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+class MetaflowGcpSecretsManagerDuplicateKey(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+class MetaflowGcpSecretsManagerJSONParseError(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+class MetaflowGcpSecretsManagerNotJSONObject(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+class GcpSecretManagerSecretsProvider(metaflow.plugins.secrets.SecretsProvider, metaclass=abc.ABCMeta):
+    def get_secret_as_dict(self, secret_id, options = {}, role = None):
+        """
+        Reads a secret from GCP Secrets Manager and returns it as a dictionary of environment variables.
+        
+        If the secret contains a string payload ("SecretString"):
+        - if the `json` option is True:
+            Secret will be parsed as a JSON. If successfully parsed, AND the JSON contains a
+            top-level object, each entry K/V in the object will also be converted to an entry in the result. V will
+            always be casted to a string (if not already a string).
+        - If `json` option is False (default):
+            Will be returned as a single entry in the result, with the key being the last part after / in secret_id.
+        
+        On GCP Secrets Manager, the secret payload is a binary blob. However, by default we interpret it as UTF8 encoded
+        string. To disable this, set the `binary` option to True, the binary will be base64 encoded in the result.
+        
+        All keys in the result are sanitized to be more valid environment variable names. This is done on a best effort
+        basis. Further validation is expected to be done by the invoking @secrets decorator itself.
+        
+        :param secret_id: GCP Secrets Manager secret ID
+        :param options: unused
+        :return: dict of environment variables. All keys and values are strings.
+        """
+        ...
+    ...
+

metaflow-stubs/plugins/gcp/gs_exceptions.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.408777                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.081366                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi

@@ -1,12 +1,26 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.408553                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.081144                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 
+class GcpDefaultClientProvider(object, metaclass=type):
+    @staticmethod
+    def get_gs_storage_client(*args, **kwargs):
+        ...
+    @staticmethod
+    def get_credentials(scopes, *args, **kwargs):
+        ...
+    ...
+
+cached_provider_class: None
+
 def get_gs_storage_client():
     ...
 
+def get_credentials(scopes, *args, **kwargs):
+    ...
+

metaflow-stubs/plugins/gcp/gs_utils.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.409140                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.081715                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/includefile_support.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.365184                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.036121                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.351848                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.023867                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/kubernetes.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.383202                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.057898                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -52,10 +52,16 @@ DATATOOLS_S3ROOT: None
 
 DEFAULT_AWS_CLIENT_PROVIDER: str
 
+DEFAULT_GCP_CLIENT_PROVIDER: str
+
 DEFAULT_METADATA: str
 
 DEFAULT_SECRETS_BACKEND_TYPE: None
 
+GCP_SECRET_MANAGER_PREFIX: None
+
+AZURE_KEY_VAULT_PREFIX: None
+
 KUBERNETES_FETCH_EC2_METADATA: bool
 
 KUBERNETES_LABELS: str
@@ -79,6 +85,8 @@ class KubernetesClient(object, metaclass=type):
         ...
     def get(self):
         ...
+    def jobset(self, **kwargs):
+        ...
     def job(self, **kwargs):
         ...
     ...
@@ -104,7 +112,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/kubernetes/kubernetes_cli.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15                                                            #
-# Generated on 2024-05-08T22:46:02.385448                                        #
+# MF version: 2.11.16                                                            #
+# Generated on 2024-05-22T00:31:20.060288                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow._vendor.click.types
+    import metaflow.exception
     import metaflow.decorators
 
 class JSONTypeClass(metaflow._vendor.click.types.ParamType, metaclass=type):
@@ -29,6 +29,10 @@ class CommandException(metaflow.exception.MetaflowException, metaclass=type):
 def sync_local_metadata_from_datastore(metadata_local_dir, task_ds):
     ...
 
+UBF_CONTROL: str
+
+UBF_TASK: str
+
 DATASTORE_LOCAL_DIR: str
 
 KUBERNETES_LABELS: str
@@ -40,7 +44,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):
         ...
@@ -52,6 +60,9 @@ class KubernetesKilledException(metaflow.exception.MetaflowException, metaclass=
 def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
     ...
 
+class KubernetesException(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
 class KubernetesDecorator(metaflow.decorators.StepDecorator, metaclass=type):
     def __init__(self, attributes = None, statically_defined = False):
         ...