ob-metaflow-stubs 3.4__py2.py3-none-any.whl → 3.6__py2.py3-none-any.whl

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

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.661463                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.043855                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/card_resolver.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.687908                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.078080                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/cards/component_serializer.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.685742                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.075898                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.plugins.cards.card_modules.components
+    import metaflow.plugins.cards.card_modules.basic
     import metaflow.plugins.cards.card_modules.card
     import metaflow.exception
-    import metaflow.plugins.cards.card_modules.basic
 
 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.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.684301                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.074559                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/catch_decorator.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.653136                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.038931                                        #
 ##################################################################################
 
 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,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.652616                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.037478                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
+    import io
     import metaflow.plugins.datatools.s3.s3
     import metaflow.exception
-    import io
 
 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.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.665107                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.046456                                        #
 ##################################################################################
 
 from __future__ import annotations

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

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.678723                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.061783                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
+    import io
     import metaflow.plugins.datatools.s3.s3
     import metaflow.exception
-    import io
 
 class RangeInfo(tuple, metaclass=type):
     @staticmethod

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

@@ -1,19 +1,19 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.639354                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.023682                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import io
-    import typing
     import metaflow.metaflow_current
+    import metaflow.datastore.inputs
+    import typing
+    import io
     import metaflow.plugins.datatools.s3.s3
     import metaflow.exception
-    import metaflow.datastore.inputs
 
 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.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.710910                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.094112                                        #
 ##################################################################################
 
 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.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.664387                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.045949                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_logger.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.657172                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.040459                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/debug_monitor.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.657613                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.040680                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/environment_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.653442                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.037911                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/events_decorator.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.655291                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.039609                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.metaflow_current
     import metaflow.decorators
+    import metaflow.metaflow_current
 
 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.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.654067                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.035638                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/frameworks/pytorch.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.683300                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.054819                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.metaflow_current
-    import metaflow.plugins.parallel_decorator
     import metaflow.decorators
+    import metaflow.plugins.parallel_decorator
+    import metaflow.metaflow_current
 
 current: metaflow.metaflow_current.Current
 

metaflow-stubs/plugins/gcp/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.656792                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.040234                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.701015                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.083033                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gs_exceptions.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.700303                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.082331                                        #
 ##################################################################################
 
 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.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.700122                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.082150                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/gcp/gs_utils.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.700590                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.082614                                        #
 ##################################################################################
 
 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.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.666107                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.047395                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.650885                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.038004                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/kubernetes.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.674222                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.062766                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -112,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, annotations = None, num_parallel = 0, attrs = {}, 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,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.675921                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.065458                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
+    import metaflow.decorators
     import metaflow._vendor.click.types
     import metaflow.exception
-    import metaflow.decorators
 
 class JSONTypeClass(metaflow._vendor.click.types.ParamType, metaclass=type):
     def convert(self, value, param, ctx):
@@ -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, annotations = None, num_parallel = 0, attrs = {}, 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):
         ...

metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.624217                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.008439                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.2+ob(v1)                                                   #
-# Generated on 2024-05-17T19:44:44.675117                                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.064598                                        #
 ##################################################################################
 
 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
 
@@ -65,10 +65,10 @@ KUBERNETES_TOLERATIONS: str
 
 KUBERNETES_SERVICE_ACCOUNT: None
 
-KUBERNETES_PORT: None
-
 KUBERNETES_SHARED_MEMORY: None
 
+KUBERNETES_PORT: None
+
 KUBERNETES_CPU: None
 
 KUBERNETES_MEMORY: None
@@ -81,8 +81,6 @@ 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:
@@ -146,6 +144,25 @@ 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):
+        ...
+    ...
+
 MAX_MEMORY_PER_TASK: None
 
 MAX_CPU_PER_TASK: None

metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi

@@ -0,0 +1,127 @@
+##################################################################################
+#                       Auto-generated Metaflow stub file                        #
+# MF version: 2.11.16.1+ob(v1)                                                   #
+# Generated on 2024-05-21T17:36:50.063662                                        #
+##################################################################################
+
+from __future__ import annotations
+
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.metaflow_current
+    import metaflow.exception
+
+current: metaflow.metaflow_current.Current
+
+class MetaflowException(Exception, metaclass=type):
+    def __init__(self, msg = "", lineno = None):
+        ...
+    def __str__(self):
+        ...
+    ...
+
+UBF_CONTROL: str
+
+UBF_TASK: str
+
+KUBERNETES_JOBSET_GROUP: str
+
+KUBERNETES_JOBSET_VERSION: str
+
+class KubernetesJobsetException(metaflow.exception.MetaflowException, metaclass=type):
+    ...
+
+def k8s_retry(deadline_seconds = 60, max_backoff = 32):
+    ...
+
+class JobsetStatus(tuple, metaclass=type):
+    @staticmethod
+    def __new__(_cls, control_pod_failed, control_exit_code, control_pod_status, control_started, control_completed, worker_pods_failed, workers_are_suspended, workers_have_started, all_jobs_are_suspended, jobset_finished, jobset_failed, status_unknown, jobset_was_terminated, some_jobs_are_running):
+        """
+        Create new instance of JobsetStatus(control_pod_failed, control_exit_code, control_pod_status, control_started, control_completed, worker_pods_failed, workers_are_suspended, workers_have_started, all_jobs_are_suspended, jobset_finished, jobset_failed, status_unknown, jobset_was_terminated, some_jobs_are_running)
+        """
+        ...
+    def __repr__(self):
+        """
+        Return a nicely formatted representation string
+        """
+        ...
+    def __getnewargs__(self):
+        """
+        Return self as a plain tuple.  Used by copy and pickle.
+        """
+        ...
+    ...
+
+class RunningJobSet(object, metaclass=type):
+    def __init__(self, client, name, namespace, group, version):
+        ...
+    def __repr__(self):
+        ...
+    def kill(self):
+        ...
+    @property
+    def id(self):
+        ...
+    @property
+    def is_done(self):
+        ...
+    @property
+    def status(self):
+        ...
+    @property
+    def has_succeeded(self):
+        ...
+    @property
+    def has_failed(self):
+        ...
+    @property
+    def is_running(self):
+        ...
+    @property
+    def _jobset_was_terminated(self):
+        ...
+    @property
+    def is_waiting(self):
+        ...
+    @property
+    def reason(self):
+        ...
+    @property
+    def _jobset_is_completed(self):
+        ...
+    @property
+    def _jobset_has_failed(self):
+        ...
+    ...
+
+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):
+        ...
+    ...
+
+def get_control_job(client, job_spec, jobset_main_addr, subdomain, port = None, num_parallel = None, namespace = None, annotations = None) -> dict:
+    ...
+
+def get_worker_job(client, job_spec, job_name, jobset_main_addr, subdomain, control_task_id = None, worker_task_id = None, replicas = 1, port = None, num_parallel = None, namespace = None, annotations = None) -> dict:
+    ...
+
+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 execute(self):
+        ...
+    ...
+