polyaxon-sdk 1.18.0rc7__py2.py3-none-any.whl → 2.1.6__py2.py3-none-any.whl

polyaxon_sdk/models/v1_run_edge_kind.py

@@ -1,134 +1,42 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
+from aenum import Enum, no_arg
 
-from polyaxon_sdk.configuration import Configuration
 
 
-class V1RunEdgeKind(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
 
-    Do not edit the class manually.
-    """
 
+class V1RunEdgeKind(str, Enum):
     """
-    allowed enum values
+    V1RunEdgeKind
     """
-    ACTION = "action"
-    EVENT = "event"
-    HOOK = "hook"
-    DAG = "dag"
-    JOIN = "join"
-    RUN = "run"
-    TB = "tb"
-    BUILD = "build"
-
-    allowable_values = [ACTION, EVENT, HOOK, DAG, JOIN, RUN, TB, BUILD]  # noqa: E501
 
     """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
+    allowed enum values
     """
-    openapi_types = {
-    }
-
-    attribute_map = {
-    }
-
-    def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """V1RunEdgeKind - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-        self.discriminator = None
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(value)
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1RunEdgeKind):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, V1RunEdgeKind):
-            return True
+    ACTION = 'action'
+    EVENT = 'event'
+    HOOK = 'hook'
+    DAG = 'dag'
+    JOIN = 'join'
+    RUN = 'run'
+    TB = 'tb'
+    BUILD = 'build'
+    MANUAL = 'manual'
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_run_edge_lineage.py

@@ -0,0 +1,75 @@
+# coding: utf-8
+
+"""
+    Polyaxon SDKs and REST API specification.
+
+         # noqa: E501
+
+    The version of the OpenAPI document: 2.1.6
+    Contact: contact@polyaxon.com
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""
+
+
+from __future__ import annotations
+from inspect import getfullargspec
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import Any, Dict, Optional
+from pydantic import BaseModel, StrictBool, StrictStr
+
+class V1RunEdgeLineage(BaseModel):
+    """
+    V1RunEdgeLineage
+    """
+    uuid: Optional[StrictStr] = None
+    is_upstream: Optional[StrictBool] = None
+    values: Optional[Dict[str, Any]] = None
+    __properties = ["uuid", "is_upstream", "values"]
+
+    class Config:
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> V1RunEdgeLineage:
+        """Create an instance of V1RunEdgeLineage from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1RunEdgeLineage:
+        """Create an instance of V1RunEdgeLineage from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1RunEdgeLineage.parse_obj(obj)
+
+        _obj = V1RunEdgeLineage.parse_obj({
+            "uuid": obj.get("uuid"),
+            "is_upstream": obj.get("is_upstream"),
+            "values": obj.get("values")
+        })
+        return _obj
+

polyaxon_sdk/models/v1_run_edges_graph.py

@@ -0,0 +1,79 @@
+# coding: utf-8
+
+"""
+    Polyaxon SDKs and REST API specification.
+
+         # noqa: E501
+
+    The version of the OpenAPI document: 2.1.6
+    Contact: contact@polyaxon.com
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""
+
+
+from __future__ import annotations
+from inspect import getfullargspec
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import List, Optional
+from pydantic import BaseModel, conlist
+from polyaxon_sdk.models.v1_run_edge_lineage import V1RunEdgeLineage
+
+class V1RunEdgesGraph(BaseModel):
+    """
+    V1RunEdgesGraph
+    """
+    edges: Optional[conlist(V1RunEdgeLineage)] = None
+    __properties = ["edges"]
+
+    class Config:
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> V1RunEdgesGraph:
+        """Create an instance of V1RunEdgesGraph from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True,
+                          exclude={
+                          },
+                          exclude_none=True)
+        # override the default output from pydantic by calling `to_dict()` of each item in edges (list)
+        _items = []
+        if self.edges:
+            for _item in self.edges:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['edges'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1RunEdgesGraph:
+        """Create an instance of V1RunEdgesGraph from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1RunEdgesGraph.parse_obj(obj)
+
+        _obj = V1RunEdgesGraph.parse_obj({
+            "edges": [V1RunEdgeLineage.from_dict(_item) for _item in obj.get("edges")] if obj.get("edges") is not None else None
+        })
+        return _obj
+

polyaxon_sdk/models/v1_run_kind.py

@@ -1,145 +1,51 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
+from aenum import Enum, no_arg
 
-from polyaxon_sdk.configuration import Configuration
 
 
-class V1RunKind(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
 
-    Do not edit the class manually.
-    """
 
+class V1RunKind(str, Enum):
     """
-    allowed enum values
+    V1RunKind
     """
-    JOB = "job"
-    SERVICE = "service"
-    DAG = "dag"
-    SPARK = "spark"
-    DASK = "dask"
-    FLINK = "flink"
-    RAY = "ray"
-    MPIJOB = "mpijob"
-    TFJOB = "tfjob"
-    PYTORCHJOB = "pytorchjob"
-    MXJOB = "mxjob"
-    XGBJOB = "xgbjob"
-    MATRIX = "matrix"
-    SCHEDULE = "schedule"
-    TUNER = "tuner"
-    WATCHDOG = "watchdog"
-    NOTIFIER = "notifier"
-    CLEANER = "cleaner"
-    BUILDER = "builder"
-
-    allowable_values = [JOB, SERVICE, DAG, SPARK, DASK, FLINK, RAY, MPIJOB, TFJOB, PYTORCHJOB, MXJOB, XGBJOB, MATRIX, SCHEDULE, TUNER, WATCHDOG, NOTIFIER, CLEANER, BUILDER]  # noqa: E501
 
     """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
+    allowed enum values
     """
-    openapi_types = {
-    }
-
-    attribute_map = {
-    }
-
-    def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """V1RunKind - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-        self.discriminator = None
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(value)
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1RunKind):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, V1RunKind):
-            return True
+    JOB = 'job'
+    SERVICE = 'service'
+    DAG = 'dag'
+    DASKJOB = 'daskjob'
+    RAYJOB = 'rayjob'
+    MPIJOB = 'mpijob'
+    TFJOB = 'tfjob'
+    PYTORCHJOB = 'pytorchjob'
+    MXJOB = 'mxjob'
+    XGBJOB = 'xgbjob'
+    PADDLEJOB = 'paddlejob'
+    MATRIX = 'matrix'
+    SCHEDULE = 'schedule'
+    TUNER = 'tuner'
+    WATCHDOG = 'watchdog'
+    NOTIFIER = 'notifier'
+    BUILDER = 'builder'
+    CLEANER = 'cleaner'
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_run_pending.py

@@ -1,130 +1,37 @@
-#!/usr/bin/python
-#
-# Copyright 2018-2022 Polyaxon, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 # coding: utf-8
 
 """
     Polyaxon SDKs and REST API specification.
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501
 
-    The version of the OpenAPI document: 1.18.0
+    The version of the OpenAPI document: 2.1.6
     Contact: contact@polyaxon.com
-    Generated by: https://openapi-generator.tech
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
 """
 
 
-try:
-    from inspect import getfullargspec
-except ImportError:
-    from inspect import getargspec as getfullargspec
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
+from aenum import Enum, no_arg
 
-from polyaxon_sdk.configuration import Configuration
 
 
-class V1RunPending(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
 
-    Do not edit the class manually.
-    """
 
+class V1RunPending(str, Enum):
     """
-    allowed enum values
+    V1RunPending
     """
-    APPROVAL = "approval"
-    UPLOAD = "upload"
-    CACHE = "cache"
-    BUILD = "build"
-
-    allowable_values = [APPROVAL, UPLOAD, CACHE, BUILD]  # noqa: E501
 
     """
-    Attributes:
-      openapi_types (dict): The key is attribute name
-                            and the value is attribute type.
-      attribute_map (dict): The key is attribute name
-                            and the value is json key in definition.
+    allowed enum values
     """
-    openapi_types = {
-    }
-
-    attribute_map = {
-    }
-
-    def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """V1RunPending - a model defined in OpenAPI"""  # noqa: E501
-        if local_vars_configuration is None:
-            local_vars_configuration = Configuration.get_default_copy()
-        self.local_vars_configuration = local_vars_configuration
-        self.discriminator = None
-
-    def to_dict(self, serialize=False):
-        """Returns the model properties as a dict"""
-        result = {}
-
-        def convert(x):
-            if hasattr(x, "to_dict"):
-                args = getfullargspec(x.to_dict).args
-                if len(args) == 1:
-                    return x.to_dict()
-                else:
-                    return x.to_dict(serialize)
-            else:
-                return x
-
-        for attr, _ in six.iteritems(self.openapi_types):
-            value = getattr(self, attr)
-            attr = self.attribute_map.get(attr, attr) if serialize else attr
-            if isinstance(value, list):
-                result[attr] = list(map(
-                    lambda x: convert(x),
-                    value
-                ))
-            elif isinstance(value, dict):
-                result[attr] = dict(map(
-                    lambda item: (item[0], convert(item[1])),
-                    value.items()
-                ))
-            else:
-                result[attr] = convert(value)
-
-        return result
-
-    def to_str(self):
-        """Returns the string representation of the model"""
-        return pprint.pformat(self.to_dict())
-
-    def __repr__(self):
-        """For `print` and `pprint`"""
-        return self.to_str()
-
-    def __eq__(self, other):
-        """Returns true if both objects are equal"""
-        if not isinstance(other, V1RunPending):
-            return False
-
-        return self.to_dict() == other.to_dict()
-
-    def __ne__(self, other):
-        """Returns true if both objects are not equal"""
-        if not isinstance(other, V1RunPending):
-            return True
+    APPROVAL = 'approval'
+    UPLOAD = 'upload'
+    CACHE = 'cache'
+    BUILD = 'build'
 
-        return self.to_dict() != other.to_dict()