ob-metaflow-stubs 6.0.3.129__py2.py3-none-any.whl → 6.0.3.131__py2.py3-none-any.whl

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.669771                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.272216                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.cards.card_modules.basic
     import metaflow.plugins.cards.card_modules.card
+    import metaflow.plugins.cards.card_modules.basic
 
 from .card import MetaflowCard as MetaflowCard
 from .card import MetaflowCardComponent as MetaflowCardComponent

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.665320                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.268859                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.666824                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.270378                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.688655                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.292586                                                            #
 ######################################################################################################
 
 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.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.688816                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.292815                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.680103                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.283246                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_resolver.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.686934                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.289914                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/component_serializer.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.690278                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.303998                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/exception.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.688267                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.291895                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/catch_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.676371                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.278803                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.674290                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.276054                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/datatools/local.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.694167                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.293653                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.694481                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.294081                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.697271                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.299227                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.exception
     import metaflow.plugins.datatools.s3.s3
+    import metaflow.exception
     import io
 
 from ....flowspec import FlowSpec as FlowSpec

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.749076                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.332503                                                            #
 ######################################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.748756                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.332200                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_logger.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.678890                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.281630                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_monitor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.679120                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.281946                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/environment_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.673046                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.275775                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/events_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.678387                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.280001                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/frameworks/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.675134                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.279463                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/frameworks/pytorch.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.698911                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.317605                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.674590                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.276325                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi

@@ -1,18 +1,18 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.698649                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.301465                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import abc
 import metaflow
+import abc
 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/gcp/gs_exceptions.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.697678                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.299817                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.697546                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.299629                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gs_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.697901                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.300169                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/includefile_support.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.698351                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.300983                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.676503                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.277261                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,5 +9,6 @@ from __future__ import annotations
 
 from . import kubernetes_client as kubernetes_client
 from . import kubernetes as kubernetes
+from . import kube_utils as kube_utils
 from . import kubernetes_decorator as kubernetes_decorator
 

metaflow-stubs/plugins/kubernetes/kube_utils.pyi

@@ -1,13 +1,21 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.767088                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.311173                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.exception
 
 from ...exception import CommandException as CommandException
+from ...exception import MetaflowException as MetaflowException
+
+class KubernetesException(metaflow.exception.MetaflowException, metaclass=type):
+    ...
 
 def parse_cli_options(flow_name, run_id, user, my_runs, echo):
     ...
@@ -18,3 +26,18 @@ def qos_requests_and_limits(qos: str, cpu: int, memory: int, storage: int):
     """
     ...
 
+def validate_kube_labels(labels: typing.Optional[typing.Dict[str, typing.Optional[str]]]) -> bool:
+    """
+    Validate label values.
+    
+    This validates the kubernetes label values.  It does not validate the keys.
+    Ideally, keys should be static and also the validation rules for keys are
+    more complex than those for values.  For full validation rules, see:
+    
+    https://kubernetes.io/docs/concepts/overview/working-with-objects/labels/#syntax-and-character-set
+    """
+    ...
+
+def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
+    ...
+

metaflow-stubs/plugins/kubernetes/kubernetes.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.700025                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.310578                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -64,8 +64,6 @@ GCP_SECRET_MANAGER_PREFIX: None
 
 KUBERNETES_FETCH_EC2_METADATA: bool
 
-KUBERNETES_LABELS: str
-
 KUBERNETES_SANDBOX_INIT_SCRIPT: None
 
 OTEL_ENDPOINT: None
@@ -109,9 +107,9 @@ class Kubernetes(object, metaclass=type):
         ...
     def launch_job(self, **kwargs):
         ...
-    def create_jobset(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, docker_image, docker_image_pull_policy, step_cli = None, 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, num_parallel = None, qos = None):
+    def create_jobset(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, docker_image, docker_image_pull_policy, step_cli = None, 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, annotations = None, shared_memory = None, port = None, num_parallel = None, qos = 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, qos = 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, qos = None, annotations = None):
         ...
     def create_k8sjob(self, job):
         ...
@@ -119,18 +117,3 @@ class Kubernetes(object, metaclass=type):
         ...
     ...
 
-def validate_kube_labels(labels: typing.Optional[typing.Dict[str, typing.Optional[str]]]) -> bool:
-    """
-    Validate label values.
-    
-    This validates the kubernetes label values.  It does not validate the keys.
-    Ideally, keys should be static and also the validation rules for keys are
-    more complex than those for values.  For full validation rules, see:
-    
-    https://kubernetes.io/docs/concepts/overview/working-with-objects/labels/#syntax-and-character-set
-    """
-    ...
-
-def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
-    ...
-

metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.632379                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.234571                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.700719                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.312335                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -20,7 +20,8 @@ from ..timeout_decorator import get_run_time_limit_for_task as get_run_time_limi
 from ..aws.aws_utils import get_docker_registry as get_docker_registry
 from ..aws.aws_utils import get_ec2_instance_metadata as get_ec2_instance_metadata
 from .kubernetes import KubernetesException as KubernetesException
-from .kubernetes import parse_kube_keyvalue_list as parse_kube_keyvalue_list
+from .kube_utils import validate_kube_labels as validate_kube_labels
+from .kube_utils import parse_kube_keyvalue_list as parse_kube_keyvalue_list
 
 DATASTORE_LOCAL_DIR: str
 
@@ -40,6 +41,10 @@ KUBERNETES_IMAGE_PULL_POLICY: None
 
 KUBERNETES_MEMORY: None
 
+KUBERNETES_LABELS: str
+
+KUBERNETES_ANNOTATIONS: str
+
 KUBERNETES_NAMESPACE: str
 
 KUBERNETES_NODE_SELECTOR: str
@@ -108,6 +113,10 @@ class KubernetesDecorator(metaflow.decorators.StepDecorator, metaclass=type):
     tolerations : List[str], default []
         The default is extracted from METAFLOW_KUBERNETES_TOLERATIONS.
         Kubernetes tolerations to use when launching pod in Kubernetes.
+    labels: Dict[str, str], default: METAFLOW_KUBERNETES_LABELS
+        Kubernetes labels to use when launching pod in Kubernetes.
+    annotations: Dict[str, str], default: METAFLOW_KUBERNETES_ANNOTATIONS
+        Kubernetes annotations to use when launching pod in Kubernetes.
     use_tmpfs : bool, default False
         This enables an explicit tmpfs mount for this step.
     tmpfs_tempdir : bool, default True

metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.768438                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.383688                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -18,8 +18,6 @@ KUBERNETES_JOBSET_GROUP: str
 
 KUBERNETES_JOBSET_VERSION: str
 
-KUBERNETES_SECRETS: str
-
 class KubernetesJobsetException(metaflow.exception.MetaflowException, metaclass=type):
     ...
 
@@ -143,6 +141,10 @@ class KubernetesJobSet(object, metaclass=type):
         ...
     def annotation(self, name, value):
         ...
+    def labels(self, labels):
+        ...
+    def annotations(self, annotations):
+        ...
     def secret(self, name):
         ...
     def dump(self):
@@ -171,8 +173,12 @@ class KubernetesArgoJobSet(object, metaclass=type):
         ...
     def label(self, name, value):
         ...
+    def labels(self, labels):
+        ...
     def annotation(self, name, value):
         ...
+    def annotations(self, annotations):
+        ...
     def dump(self):
         ...
     ...

metaflow-stubs/plugins/parallel_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.675634                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.275609                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/perimeters.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.633033                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.235247                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/project_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.677942                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.280519                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/pypi/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1)                                                    #
-# Generated on 2024-12-20T17:11:09.676976                                                            #
+# MF version: 2.13.2.1+obcheckpoint(0.1.6);ob(v1)                                                    #
+# Generated on 2025-01-08T23:52:28.277076                                                            #
 ######################################################################################################
 
 from __future__ import annotations