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

polyaxon_sdk/models/v1_early_stopping.py

@@ -1,252 +1,99 @@
-#!/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 __future__ import annotations
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
-
-from polyaxon_sdk.configuration import Configuration
+import json
 
 
-class V1EarlyStopping(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from typing import Optional
+from pydantic import BaseModel
+from polyaxon_sdk.models.v1_diff_stopping_policy import V1DiffStoppingPolicy
+from polyaxon_sdk.models.v1_failure_early_stopping import V1FailureEarlyStopping
+from polyaxon_sdk.models.v1_median_stopping_policy import V1MedianStoppingPolicy
+from polyaxon_sdk.models.v1_metric_early_stopping import V1MetricEarlyStopping
+from polyaxon_sdk.models.v1_truncation_stopping_policy import V1TruncationStoppingPolicy
 
+class V1EarlyStopping(BaseModel):
     """
-    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.
+    V1EarlyStopping
     """
-    openapi_types = {
-        'median': 'V1MedianStoppingPolicy',
-        'diff': 'V1DiffStoppingPolicy',
-        'truncation': 'V1TruncationStoppingPolicy',
-        'metric': 'V1MetricEarlyStopping',
-        'failure': 'V1FailureEarlyStopping'
-    }
-
-    attribute_map = {
-        'median': 'median',
-        'diff': 'diff',
-        'truncation': 'truncation',
-        'metric': 'metric',
-        'failure': 'failure'
-    }
-
-    def __init__(self, median=None, diff=None, truncation=None, metric=None, failure=None, local_vars_configuration=None):  # noqa: E501
-        """V1EarlyStopping - 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._median = None
-        self._diff = None
-        self._truncation = None
-        self._metric = None
-        self._failure = None
-        self.discriminator = None
-
-        if median is not None:
-            self.median = median
-        if diff is not None:
-            self.diff = diff
-        if truncation is not None:
-            self.truncation = truncation
-        if metric is not None:
-            self.metric = metric
-        if failure is not None:
-            self.failure = failure
-
-    @property
-    def median(self):
-        """Gets the median of this V1EarlyStopping.  # noqa: E501
-
-
-        :return: The median of this V1EarlyStopping.  # noqa: E501
-        :rtype: V1MedianStoppingPolicy
-        """
-        return self._median
-
-    @median.setter
-    def median(self, median):
-        """Sets the median of this V1EarlyStopping.
-
-
-        :param median: The median of this V1EarlyStopping.  # noqa: E501
-        :type median: V1MedianStoppingPolicy
-        """
-
-        self._median = median
-
-    @property
-    def diff(self):
-        """Gets the diff of this V1EarlyStopping.  # noqa: E501
-
-
-        :return: The diff of this V1EarlyStopping.  # noqa: E501
-        :rtype: V1DiffStoppingPolicy
-        """
-        return self._diff
-
-    @diff.setter
-    def diff(self, diff):
-        """Sets the diff of this V1EarlyStopping.
-
-
-        :param diff: The diff of this V1EarlyStopping.  # noqa: E501
-        :type diff: V1DiffStoppingPolicy
-        """
-
-        self._diff = diff
-
-    @property
-    def truncation(self):
-        """Gets the truncation of this V1EarlyStopping.  # noqa: E501
-
-
-        :return: The truncation of this V1EarlyStopping.  # noqa: E501
-        :rtype: V1TruncationStoppingPolicy
-        """
-        return self._truncation
-
-    @truncation.setter
-    def truncation(self, truncation):
-        """Sets the truncation of this V1EarlyStopping.
-
-
-        :param truncation: The truncation of this V1EarlyStopping.  # noqa: E501
-        :type truncation: V1TruncationStoppingPolicy
-        """
-
-        self._truncation = truncation
-
-    @property
-    def metric(self):
-        """Gets the metric of this V1EarlyStopping.  # noqa: E501
-
-
-        :return: The metric of this V1EarlyStopping.  # noqa: E501
-        :rtype: V1MetricEarlyStopping
-        """
-        return self._metric
-
-    @metric.setter
-    def metric(self, metric):
-        """Sets the metric of this V1EarlyStopping.
-
-
-        :param metric: The metric of this V1EarlyStopping.  # noqa: E501
-        :type metric: V1MetricEarlyStopping
-        """
-
-        self._metric = metric
-
-    @property
-    def failure(self):
-        """Gets the failure of this V1EarlyStopping.  # noqa: E501
-
-
-        :return: The failure of this V1EarlyStopping.  # noqa: E501
-        :rtype: V1FailureEarlyStopping
-        """
-        return self._failure
-
-    @failure.setter
-    def failure(self, failure):
-        """Sets the failure of this V1EarlyStopping.
-
-
-        :param failure: The failure of this V1EarlyStopping.  # noqa: E501
-        :type failure: V1FailureEarlyStopping
-        """
-
-        self._failure = failure
-
-    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, V1EarlyStopping):
-            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, V1EarlyStopping):
-            return True
+    median: Optional[V1MedianStoppingPolicy] = None
+    diff: Optional[V1DiffStoppingPolicy] = None
+    truncation: Optional[V1TruncationStoppingPolicy] = None
+    metric: Optional[V1MetricEarlyStopping] = None
+    failure: Optional[V1FailureEarlyStopping] = None
+    __properties = ["median", "diff", "truncation", "metric", "failure"]
+
+    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) -> V1EarlyStopping:
+        """Create an instance of V1EarlyStopping 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 median
+        if self.median:
+            _dict['median'] = self.median.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of diff
+        if self.diff:
+            _dict['diff'] = self.diff.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of truncation
+        if self.truncation:
+            _dict['truncation'] = self.truncation.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of metric
+        if self.metric:
+            _dict['metric'] = self.metric.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of failure
+        if self.failure:
+            _dict['failure'] = self.failure.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1EarlyStopping:
+        """Create an instance of V1EarlyStopping from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1EarlyStopping.parse_obj(obj)
+
+        _obj = V1EarlyStopping.parse_obj({
+            "median": V1MedianStoppingPolicy.from_dict(obj.get("median")) if obj.get("median") is not None else None,
+            "diff": V1DiffStoppingPolicy.from_dict(obj.get("diff")) if obj.get("diff") is not None else None,
+            "truncation": V1TruncationStoppingPolicy.from_dict(obj.get("truncation")) if obj.get("truncation") is not None else None,
+            "metric": V1MetricEarlyStopping.from_dict(obj.get("metric")) if obj.get("metric") is not None else None,
+            "failure": V1FailureEarlyStopping.from_dict(obj.get("failure")) if obj.get("failure") is not None else None
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_entities_tags.py

@@ -1,174 +1,73 @@
-#!/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 __future__ import annotations
+from inspect import getfullargspec
 import pprint
 import re  # noqa: F401
-import six
-
-from polyaxon_sdk.configuration import Configuration
+import json
 
 
-class V1EntitiesTags(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from typing import List, Optional
+from pydantic import BaseModel, StrictStr, conlist
 
+class V1EntitiesTags(BaseModel):
     """
-    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.
+    V1EntitiesTags
     """
-    openapi_types = {
-        'uuids': 'list[str]',
-        'tags': 'list[str]'
-    }
-
-    attribute_map = {
-        'uuids': 'uuids',
-        'tags': 'tags'
-    }
-
-    def __init__(self, uuids=None, tags=None, local_vars_configuration=None):  # noqa: E501
-        """V1EntitiesTags - 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._uuids = None
-        self._tags = None
-        self.discriminator = None
-
-        if uuids is not None:
-            self.uuids = uuids
-        if tags is not None:
-            self.tags = tags
-
-    @property
-    def uuids(self):
-        """Gets the uuids of this V1EntitiesTags.  # noqa: E501
-
-
-        :return: The uuids of this V1EntitiesTags.  # noqa: E501
-        :rtype: list[str]
-        """
-        return self._uuids
-
-    @uuids.setter
-    def uuids(self, uuids):
-        """Sets the uuids of this V1EntitiesTags.
-
-
-        :param uuids: The uuids of this V1EntitiesTags.  # noqa: E501
-        :type uuids: list[str]
-        """
-
-        self._uuids = uuids
-
-    @property
-    def tags(self):
-        """Gets the tags of this V1EntitiesTags.  # noqa: E501
-
-
-        :return: The tags of this V1EntitiesTags.  # noqa: E501
-        :rtype: list[str]
-        """
-        return self._tags
-
-    @tags.setter
-    def tags(self, tags):
-        """Sets the tags of this V1EntitiesTags.
-
-
-        :param tags: The tags of this V1EntitiesTags.  # noqa: E501
-        :type tags: list[str]
-        """
-
-        self._tags = tags
-
-    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, V1EntitiesTags):
-            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, V1EntitiesTags):
-            return True
+    uuids: Optional[conlist(StrictStr)] = None
+    tags: Optional[conlist(StrictStr)] = None
+    __properties = ["uuids", "tags"]
+
+    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) -> V1EntitiesTags:
+        """Create an instance of V1EntitiesTags 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) -> V1EntitiesTags:
+        """Create an instance of V1EntitiesTags from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1EntitiesTags.parse_obj(obj)
+
+        _obj = V1EntitiesTags.parse_obj({
+            "uuids": obj.get("uuids"),
+            "tags": obj.get("tags")
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()