metaflow-stubs 2.12.27__py2.py3-none-any.whl → 2.12.29__py2.py3-none-any.whl

metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi

@@ -1,50 +1,19 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27                                                            #
-# Generated on 2024-10-24T19:27:18.000385                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.644829                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.plugins.airflow.sensors.base_sensor
-    import metaflow.exception
-    import metaflow.decorators
 
-class AirflowSensorDecorator(metaflow.decorators.FlowDecorator, metaclass=type):
-    """
-    Base class for all Airflow sensor decorators.
-    """
-    def __init__(self, *args, **kwargs):
-        ...
-    def serialize_operator_args(self):
-        """
-        Subclasses will parse the decorator arguments to
-        Airflow task serializable arguments.
-        """
-        ...
-    def create_task(self):
-        ...
-    def validate(self, flow):
-        """
-        Validate if the arguments for the sensor are correct.
-        """
-        ...
-    def flow_init(self, flow, graph, environment, flow_datastore, metadata, logger, echo, options):
-        ...
-    ...
-
-class SensorNames(object, metaclass=type):
-    @classmethod
-    def get_supported_sensors(cls):
-        ...
-    ...
-
-class AirflowException(metaflow.exception.MetaflowException, metaclass=type):
-    def __init__(self, msg):
-        ...
-    ...
+from .base_sensor import AirflowSensorDecorator as AirflowSensorDecorator
+from ..airflow_utils import SensorNames as SensorNames
+from ..exception import AirflowException as AirflowException
 
 class S3KeySensorDecorator(metaflow.plugins.airflow.sensors.base_sensor.AirflowSensorDecorator, metaclass=type):
     """

metaflow-stubs/plugins/argo/__init__.pyi

@@ -1,9 +1,16 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27                                                            #
-# Generated on 2024-10-24T19:27:17.919337                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.584185                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
+from . import argo_events as argo_events
+from . import argo_workflows_decorator as argo_workflows_decorator
+from . import argo_workflows_deployer as argo_workflows_deployer
+from . import argo_client as argo_client
+from . import argo_workflows as argo_workflows
+from . import argo_workflows_cli as argo_workflows_cli
+

metaflow-stubs/plugins/argo/argo_client.pyi

@@ -1,36 +1,18 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27                                                            #
-# Generated on 2024-10-24T19:27:17.961932                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.605508                                                            #
+######################################################################################################
 
 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 jobset(self, **kwargs):
-        ...
-    def job(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                                                            #
-# Generated on 2024-10-24T19:27:17.959562                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.604019                                                            #
+######################################################################################################
 
 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,28 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27                                                            #
-# Generated on 2024-10-24T19:27:17.967660                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.610522                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.graph
-    import metaflow._vendor.click.types
-    import metaflow.parameters
     import metaflow.exception
-    import metaflow.metaflow_current
-
-inf: float
-
-JSONType: metaflow.parameters.JSONTypeClass
+    import metaflow.graph
 
-current: metaflow.metaflow_current.Current
+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 ...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 MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
-
-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,100 +94,10 @@ UI_URL: None
 
 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):
     ...