metaflow-stubs 2.12.8__py2.py3-none-any.whl → 2.12.9__py2.py3-none-any.whl

metaflow-stubs/plugins/catch_decorator.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.384924                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.272280                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.metaflow_current
     import metaflow.exception
+    import metaflow.decorators
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/plugins/datatools/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.386564                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.271242                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.plugins.datatools.s3.s3
-    import metaflow.exception
     import io
+    import metaflow.exception
 
 def read_in_chunks(dst, src, src_sz, max_chunk_size):
     ...

metaflow-stubs/plugins/datatools/local.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.397766                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.266837                                        #
 ##################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.410482                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.309207                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.plugins.datatools.s3.s3
-    import metaflow.exception
     import io
+    import metaflow.exception
 
 class RangeInfo(tuple, metaclass=type):
     @staticmethod

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

@@ -1,20 +1,20 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.365815                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.246161                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.datatools.s3.s3
-    import metaflow.datastore.inputs
+    import io
     import metaflow.flowspec
+    import typing
     import metaflow.metaflow_current
+    import metaflow.datastore.inputs
+    import metaflow.plugins.datatools.s3.s3
     import metaflow.exception
-    import typing
-    import io
 
 TYPE_CHECKING: bool
 

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

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.459274                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.349088                                        #
 ##################################################################################
 
 from __future__ import annotations

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

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.397143                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.282648                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_logger.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.390333                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.275824                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_monitor.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.390602                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.276128                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/environment_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.386930                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.274107                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/events_decorator.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.388968                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.274493                                        #
 ##################################################################################
 
 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.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.387283                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.273214                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/frameworks/pytorch.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.423253                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.331520                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.metaflow_current
     import metaflow.plugins.parallel_decorator
+    import metaflow.decorators
 
 current: metaflow.metaflow_current.Current
 
@@ -21,6 +21,8 @@ class ParallelDecorator(metaflow.decorators.StepDecorator, metaclass=type):
         ...
     def step_init(self, flow, graph, step_name, decorators, environment, flow_datastore, logger):
         ...
+    def task_pre_step(self, step_name, task_datastore, metadata, run_id, task_id, flow, graph, retry_count, max_user_code_retries, ubf_context, inputs):
+        ...
     def task_decorate(self, step_func, flow, graph, retry_count, max_user_code_retries, ubf_context):
         ...
     def setup_distributed_env(self, flow):

metaflow-stubs/plugins/gcp/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.389537                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.275562                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.445518                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.334837                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import abc
-    import metaflow.exception
     import metaflow.plugins.secrets
+    import metaflow.exception
+    import abc
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/plugins/gcp/gs_exceptions.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.444546                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.333986                                        #
 ##################################################################################
 
 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.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.444324                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.333770                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gs_utils.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.445008                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.334327                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/includefile_support.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.398884                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.268027                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.388299                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.268961                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/kubernetes.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.441756                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.294746                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -26,13 +26,15 @@ ARGO_EVENTS_EVENT_BUS: str
 
 ARGO_EVENTS_EVENT_SOURCE: None
 
+ARGO_EVENTS_INTERNAL_WEBHOOK_URL: None
+
 ARGO_EVENTS_SERVICE_ACCOUNT: None
 
-ARGO_EVENTS_INTERNAL_WEBHOOK_URL: None
+ARGO_EVENTS_WEBHOOK_AUTH: str
 
 AWS_SECRETS_MANAGER_DEFAULT_REGION: None
 
-ARGO_EVENTS_WEBHOOK_AUTH: str
+AZURE_KEY_VAULT_PREFIX: None
 
 AZURE_STORAGE_BLOB_SERVICE_ENDPOINT: None
 
@@ -60,23 +62,27 @@ DEFAULT_SECRETS_BACKEND_TYPE: None
 
 GCP_SECRET_MANAGER_PREFIX: None
 
-AZURE_KEY_VAULT_PREFIX: None
-
 KUBERNETES_FETCH_EC2_METADATA: bool
 
 KUBERNETES_LABELS: str
 
 KUBERNETES_SANDBOX_INIT_SCRIPT: None
 
+OTEL_ENDPOINT: None
+
 S3_ENDPOINT_URL: None
 
+S3_SERVER_SIDE_ENCRYPTION: None
+
 SERVICE_HEADERS: dict
 
+KUBERNETES_SECRETS: str
+
 SERVICE_INTERNAL_URL: None
 
-S3_SERVER_SIDE_ENCRYPTION: None
+UBF_CONTROL: str
 
-OTEL_ENDPOINT: None
+UBF_TASK: str
 
 BASH_SAVE_LOGS: str
 
@@ -101,6 +107,8 @@ STDOUT_PATH: str
 
 STDERR_PATH: str
 
+METAFLOW_PARALLEL_STEP_CLI_OPTIONS_TEMPLATE: str
+
 class KubernetesException(metaflow.exception.MetaflowException, metaclass=type):
     ...
 
@@ -112,9 +120,9 @@ class Kubernetes(object, metaclass=type):
         ...
     def launch_job(self, **kwargs):
         ...
-    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_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):
         ...
-    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_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):
         ...
     def create_k8sjob(self, job):
         ...

metaflow-stubs/plugins/kubernetes/kubernetes_cli.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.444012                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.297011                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
+    import metaflow.exception
     import metaflow._vendor.click.types
     import metaflow.decorators
-    import metaflow.exception
 
 class JSONTypeClass(metaflow._vendor.click.types.ParamType, metaclass=type):
     def convert(self, value, param, ctx):
@@ -29,24 +29,24 @@ 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
 
 TASK_LOG_SOURCE: str
 
+UBF_CONTROL: str
+
+UBF_TASK: str
+
 class Kubernetes(object, metaclass=type):
     def __init__(self, datastore, metadata, environment):
         ...
     def launch_job(self, **kwargs):
         ...
-    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_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):
         ...
-    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_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):
         ...
     def create_k8sjob(self, job):
         ...
@@ -54,13 +54,13 @@ class Kubernetes(object, metaclass=type):
         ...
     ...
 
-class KubernetesKilledException(metaflow.exception.MetaflowException, metaclass=type):
+class KubernetesException(metaflow.exception.MetaflowException, metaclass=type):
     ...
 
-def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
+class KubernetesKilledException(metaflow.exception.MetaflowException, metaclass=type):
     ...
 
-class KubernetesException(metaflow.exception.MetaflowException, metaclass=type):
+def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
     ...
 
 class KubernetesDecorator(metaflow.decorators.StepDecorator, metaclass=type):

metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.440555                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.293425                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -49,7 +49,30 @@ class KubernetesJob(object, metaclass=type):
     ...
 
 class KubernetesJobSet(object, metaclass=type):
-    def __init__(self, client, name = None, job_spec = None, namespace = None, num_parallel = None, annotations = None, labels = None, port = None, task_id = None, **kwargs):
+    def __init__(self, client, name = None, namespace = None, num_parallel = None, **kwargs):
+        ...
+    @property
+    def jobset_control_addr(self):
+        ...
+    @property
+    def worker(self):
+        ...
+    @property
+    def control(self):
+        ...
+    def environment_variable_from_selector(self, name, label_value):
+        ...
+    def environment_variables_from_selectors(self, env_dict):
+        ...
+    def environment_variable(self, name, value):
+        ...
+    def label(self, name, value):
+        ...
+    def annotation(self, name, value):
+        ...
+    def secret(self, name):
+        ...
+    def dump(self):
         ...
     def execute(self):
         ...

metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.442982                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.295855                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.metaflow_current
     import metaflow.exception
+    import metaflow.decorators
 
 current: metaflow.metaflow_current.Current
 
@@ -49,11 +49,17 @@ KUBERNETES_CONTAINER_IMAGE: None
 
 KUBERNETES_CONTAINER_REGISTRY: None
 
+KUBERNETES_CPU: None
+
+KUBERNETES_DISK: None
+
 KUBERNETES_FETCH_EC2_METADATA: bool
 
+KUBERNETES_GPU_VENDOR: str
+
 KUBERNETES_IMAGE_PULL_POLICY: None
 
-KUBERNETES_GPU_VENDOR: str
+KUBERNETES_MEMORY: None
 
 KUBERNETES_NAMESPACE: str
 
@@ -61,19 +67,13 @@ KUBERNETES_NODE_SELECTOR: str
 
 KUBERNETES_PERSISTENT_VOLUME_CLAIMS: str
 
-KUBERNETES_TOLERATIONS: str
+KUBERNETES_PORT: None
 
 KUBERNETES_SERVICE_ACCOUNT: None
 
 KUBERNETES_SHARED_MEMORY: None
 
-KUBERNETES_PORT: None
-
-KUBERNETES_CPU: None
-
-KUBERNETES_MEMORY: None
-
-KUBERNETES_DISK: None
+KUBERNETES_TOLERATIONS: str
 
 class ResourcesDecorator(metaflow.decorators.StepDecorator, metaclass=type):
     ...
@@ -81,6 +81,8 @@ class ResourcesDecorator(metaflow.decorators.StepDecorator, metaclass=type):
 def get_run_time_limit_for_task(step_decos):
     ...
 
+UBF_CONTROL: str
+
 def get_docker_registry(image_uri):
     """
     Explanation:
@@ -144,25 +146,6 @@ class KubernetesException(metaflow.exception.MetaflowException, metaclass=type):
 def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
     ...
 
-UBF_CONTROL: str
-
-class TaskIdConstructor(object, metaclass=type):
-    @classmethod
-    def jobset_worker_id(cls, control_task_id: str):
-        ...
-    @classmethod
-    def join_step_task_ids(cls, num_parallel):
-        """
-        Called within the step decorator to set the `flow._control_mapper_tasks`.
-        Setting these allows the flow to know which tasks are needed in the join step.
-        We set this in the `task_pre_step` method of the decorator.
-        """
-        ...
-    @classmethod
-    def argo(cls):
-        ...
-    ...
-
 class KubernetesDecorator(metaflow.decorators.StepDecorator, metaclass=type):
     def __init__(self, attributes = None, statically_defined = False):
         ...

metaflow-stubs/plugins/kubernetes/kubernetes_job.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.8                                                             #
-# Generated on 2024-07-16T15:51:55.439853                                        #
+# MF version: 2.12.9                                                             #
+# Generated on 2024-07-25T18:20:16.292529                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -26,7 +26,30 @@ UBF_TASK: str
 CLIENT_REFRESH_INTERVAL_SECONDS: int
 
 class KubernetesJobSet(object, metaclass=type):
-    def __init__(self, client, name = None, job_spec = None, namespace = None, num_parallel = None, annotations = None, labels = None, port = None, task_id = None, **kwargs):
+    def __init__(self, client, name = None, namespace = None, num_parallel = None, **kwargs):
+        ...
+    @property
+    def jobset_control_addr(self):
+        ...
+    @property
+    def worker(self):
+        ...
+    @property
+    def control(self):
+        ...
+    def environment_variable_from_selector(self, name, label_value):
+        ...
+    def environment_variables_from_selectors(self, env_dict):
+        ...
+    def environment_variable(self, name, value):
+        ...
+    def label(self, name, value):
+        ...
+    def annotation(self, name, value):
+        ...
+    def secret(self, name):
+        ...
+    def dump(self):
         ...
     def execute(self):
         ...