ob-metaflow-stubs 6.0.3.117__py2.py3-none-any.whl → 6.0.3.119__py2.py3-none-any.whl

metaflow-stubs/plugins/argo/argo_client.pyi

@@ -1,36 +1,18 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.409372                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.2+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-21T22:12:20.854044                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
 
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
-
-class KubernetesClient(object, metaclass=type):
-    def __init__(self):
-        ...
-    def get(self):
-        ...
-    def list(self, flow_name, run_id, user):
-        ...
-    def kill_pods(self, flow_name, run_id, user, echo):
-        ...
-    def job(self, **kwargs):
-        ...
-    def jobset(self, **kwargs):
-        ...
-    ...
+from ...exception import MetaflowException as MetaflowException
+from ..kubernetes.kubernetes_client import KubernetesClient as KubernetesClient
 
 class ArgoClientException(metaflow.exception.MetaflowException, metaclass=type):
     ...

metaflow-stubs/plugins/argo/argo_events.pyi

@@ -1,21 +1,17 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.407375                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.2+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-21T22:12:20.852834                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
 
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from ...exception import MetaflowException as MetaflowException
 
 ARGO_EVENTS_WEBHOOK_AUTH: str
 

metaflow-stubs/plugins/argo/argo_workflows.pyi

@@ -1,42 +1,29 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.413943                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.2+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-21T22:12:20.858042                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.graph
-    import metaflow.parameters
-    import metaflow._vendor.click.types
-    import metaflow.metaflow_current
     import metaflow.exception
 
-inf: float
-
-JSONType: metaflow.parameters.JSONTypeClass
-
-current: metaflow.metaflow_current.Current
-
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from ...parameters import JSONType as JSONType
+from ...metaflow_current import current as current
+from ...exception import MetaflowException as MetaflowException
+from ...includefile import FilePathClass as FilePathClass
+from ...mf_extensions.outerbounds.remote_config import init_config as init_config
+from ...parameters import deploy_time_eval as deploy_time_eval
+from ..kubernetes.kubernetes import parse_kube_keyvalue_list as parse_kube_keyvalue_list
+from ..kubernetes.kubernetes import validate_kube_labels as validate_kube_labels
+from ..kubernetes.kubernetes_jobsets import KubernetesArgoJobSet as KubernetesArgoJobSet
+from .argo_client import ArgoClient as ArgoClient
 
-class FilePathClass(metaflow._vendor.click.types.ParamType, metaclass=type):
-    def __init__(self, is_text, encoding):
-        ...
-    def convert(self, value, param, ctx):
-        ...
-    def __str__(self):
-        ...
-    def __repr__(self):
-        ...
-    ...
+inf: float
 
 ARGO_EVENTS_EVENT: None
 
@@ -108,111 +95,12 @@ UI_URL: None
 
 PAGERDUTY_TEMPLATE_URL: None
 
-def init_config() -> typing.Dict[str, str]:
-    """
-    OSS Metaflow reads the config file on every step initialization. This is because OSS assumes config files change
-    relatively infrequently. We want to avoid config values changing between flow steps. Our solution to prevent this
-    is to read a config once and cache it on an environment variable. Environment variables carry over between steps
-    because steps are executed in subprocesses (local) or environments which expect environment variables to be set.
-    """
-    ...
-
 BASH_SAVE_LOGS: str
 
-def deploy_time_eval(value):
-    ...
-
-def parse_kube_keyvalue_list(items: typing.List[str], requires_both: bool = True):
-    ...
-
-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
-    """
-    ...
-
-class KubernetesArgoJobSet(object, metaclass=type):
-    def __init__(self, kubernetes_sdk, name = None, namespace = 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 dump(self):
-        ...
-    ...
-
 UBF_CONTROL: str
 
 UBF_TASK: str
 
-class ArgoClient(object, metaclass=type):
-    def __init__(self, namespace = None):
-        ...
-    def get_workflow(self, name):
-        ...
-    def get_workflow_template(self, name):
-        ...
-    def get_workflow_templates(self):
-        ...
-    def register_workflow_template(self, name, workflow_template):
-        ...
-    def delete_cronworkflow(self, name):
-        """
-        Issues an API call for deleting a cronworkflow
-        
-        Returns either the successful API response, or None in case the resource was not found.
-        """
-        ...
-    def delete_workflow_template(self, name):
-        """
-        Issues an API call for deleting a cronworkflow
-        
-        Returns either the successful API response, or None in case the resource was not found.
-        """
-        ...
-    def terminate_workflow(self, name):
-        ...
-    def suspend_workflow(self, name):
-        ...
-    def unsuspend_workflow(self, name):
-        ...
-    def trigger_workflow_template(self, name, parameters = {}):
-        ...
-    def schedule_workflow_template(self, name, schedule = None, timezone = None):
-        ...
-    def register_sensor(self, name, sensor = None):
-        ...
-    def delete_sensor(self, name):
-        """
-        Issues an API call for deleting a sensor
-        
-        Returns either the successful API response, or None in case the resource was not found.
-        """
-        ...
-    ...
-
 class ArgoWorkflowsException(metaflow.exception.MetaflowException, metaclass=type):
     ...