ob-metaflow-stubs 3.5__py2.py3-none-any.whl → 3.7__py2.py3-none-any.whl

metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi

@@ -0,0 +1,127 @@
+##################################################################################
+#                       Auto-generated Metaflow stub file                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.555178                                        #
+##################################################################################
+
+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):
+        ...
+    ...
+

metaflow-stubs/plugins/logs_cli.pyi

@@ -1,16 +1,16 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.442639                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.527944                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow._vendor.click.core
     import metaflow.plugins.logs_cli
     import metaflow.exception
+    import metaflow._vendor.click.core
 
 LOGGER_TIMESTAMP: str
 

metaflow-stubs/plugins/package_cli.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.443958                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.527525                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/parallel_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.432988                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.521532                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/perimeters.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.407159                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.491600                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/project_decorator.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.437478                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.523138                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.metaflow_current
+    import metaflow.decorators
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/plugins/pypi/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.433428                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.518763                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/pypi/conda_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.456935                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.547813                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/pypi/conda_environment.pyi

@@ -1,17 +1,17 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.458135                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.549152                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import abc
     import io
-    import metaflow.metaflow_environment
     import metaflow.exception
+    import abc
+    import metaflow.metaflow_environment
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/plugins/pypi/pypi_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.456403                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.548063                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/pypi/pypi_environment.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.458525                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.549557                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/pypi/utils.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.457227                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.548349                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/resources_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.433559                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.518335                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/retry_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.436472                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.522494                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/secrets/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.436685                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.520931                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.478286                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.553229                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/secrets/secrets_decorator.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.478049                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.552974                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/storage_executor.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.437970                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.523603                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/tag_cli.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.443780                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.529167                                        #
 ##################################################################################
 
 from __future__ import annotations
@@ -9,10 +9,10 @@ from __future__ import annotations
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
-    import metaflow.events
-    import datetime
     import metaflow.client.core
     import metaflow.metaflow_current
+    import metaflow.events
+    import datetime
 
 def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:
     """

metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.432659                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.517834                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.unbounded_foreach
+    import metaflow.decorators
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):

metaflow-stubs/plugins/timeout_decorator.pyi

@@ -1,15 +1,15 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.433982                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.518213                                        #
 ##################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.decorators
     import metaflow.exception
+    import metaflow.decorators
 
 class MetaflowException(Exception, metaclass=type):
     def __init__(self, msg = "", lineno = None):
@@ -20,6 +20,8 @@ class MetaflowException(Exception, metaclass=type):
 
 UBF_CONTROL: str
 
+DEFAULT_RUNTIME_LIMIT: int
+
 class TimeoutException(metaflow.exception.MetaflowException, metaclass=type):
     ...
 

metaflow-stubs/procpoll.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.429992                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.515112                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/profilers/__init__.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.407392                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.491830                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/pylint_wrapper.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.430285                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.515393                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/tagging_util.pyi

@@ -1,7 +1,7 @@
 ##################################################################################
 #                       Auto-generated Metaflow stub file                        #
-# MF version: 2.11.15.3+ob(v1)                                                   #
-# Generated on 2024-05-17T23:07:04.410906                                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.495385                                        #
 ##################################################################################
 
 from __future__ import annotations

metaflow-stubs/tuple_util.pyi

@@ -0,0 +1,14 @@
+##################################################################################
+#                       Auto-generated Metaflow stub file                        #
+# MF version: 2.11.16.2+ob(v1)                                                   #
+# Generated on 2024-05-22T22:08:54.501475                                        #
+##################################################################################
+
+from __future__ import annotations
+
+
+def namedtuple_with_defaults(typename, field_descr, defaults = ()):
+    ...
+
+foreach_frame_field_list: list
+

{ob_metaflow_stubs-3.5.dist-info → ob_metaflow_stubs-3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-stubs
-Version: 3.5
+Version: 3.7
 Summary: Metaflow Stubs: Stubs for the metaflow package
 Author: Netflix, Outerbounds & the Metaflow Community
 Author-email: help@outerbounds.co