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

polyaxon_sdk/models/v1_truncation_stopping_policy.py

@@ -1,282 +1,81 @@
-#!/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 V1TruncationStoppingPolicy(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, Field, StrictBool, StrictInt, StrictStr
 
+class V1TruncationStoppingPolicy(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.
+    Early stopping with truncation stopping, this policy stops a percentage of all running runs at every evaluation.
     """
-    openapi_types = {
-        'kind': 'str',
-        'percent': 'int',
-        'evaluation_interval': 'int',
-        'min_interval': 'int',
-        'min_samples': 'int',
-        'include_succeeded': 'bool'
-    }
-
-    attribute_map = {
-        'kind': 'kind',
-        'percent': 'percent',
-        'evaluation_interval': 'evaluationInterval',
-        'min_interval': 'minInterval',
-        'min_samples': 'minSamples',
-        'include_succeeded': 'includeSucceeded'
-    }
-
-    def __init__(self, kind='truncation', percent=None, evaluation_interval=None, min_interval=None, min_samples=None, include_succeeded=None, local_vars_configuration=None):  # noqa: E501
-        """V1TruncationStoppingPolicy - 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._kind = None
-        self._percent = None
-        self._evaluation_interval = None
-        self._min_interval = None
-        self._min_samples = None
-        self._include_succeeded = None
-        self.discriminator = None
-
-        if kind is not None:
-            self.kind = kind
-        if percent is not None:
-            self.percent = percent
-        if evaluation_interval is not None:
-            self.evaluation_interval = evaluation_interval
-        if min_interval is not None:
-            self.min_interval = min_interval
-        if min_samples is not None:
-            self.min_samples = min_samples
-        if include_succeeded is not None:
-            self.include_succeeded = include_succeeded
-
-    @property
-    def kind(self):
-        """Gets the kind of this V1TruncationStoppingPolicy.  # noqa: E501
-
-
-        :return: The kind of this V1TruncationStoppingPolicy.  # noqa: E501
-        :rtype: str
-        """
-        return self._kind
-
-    @kind.setter
-    def kind(self, kind):
-        """Sets the kind of this V1TruncationStoppingPolicy.
-
-
-        :param kind: The kind of this V1TruncationStoppingPolicy.  # noqa: E501
-        :type kind: str
-        """
-
-        self._kind = kind
-
-    @property
-    def percent(self):
-        """Gets the percent of this V1TruncationStoppingPolicy.  # noqa: E501
-
-        The percentage of runs to stop, at each evaluation interval. e.g. 1 - 99.  # noqa: E501
-
-        :return: The percent of this V1TruncationStoppingPolicy.  # noqa: E501
-        :rtype: int
-        """
-        return self._percent
-
-    @percent.setter
-    def percent(self, percent):
-        """Sets the percent of this V1TruncationStoppingPolicy.
-
-        The percentage of runs to stop, at each evaluation interval. e.g. 1 - 99.  # noqa: E501
-
-        :param percent: The percent of this V1TruncationStoppingPolicy.  # noqa: E501
-        :type percent: int
-        """
-
-        self._percent = percent
-
-    @property
-    def evaluation_interval(self):
-        """Gets the evaluation_interval of this V1TruncationStoppingPolicy.  # noqa: E501
-
-        Interval/Frequency for applying the policy.  # noqa: E501
-
-        :return: The evaluation_interval of this V1TruncationStoppingPolicy.  # noqa: E501
-        :rtype: int
-        """
-        return self._evaluation_interval
-
-    @evaluation_interval.setter
-    def evaluation_interval(self, evaluation_interval):
-        """Sets the evaluation_interval of this V1TruncationStoppingPolicy.
-
-        Interval/Frequency for applying the policy.  # noqa: E501
-
-        :param evaluation_interval: The evaluation_interval of this V1TruncationStoppingPolicy.  # noqa: E501
-        :type evaluation_interval: int
-        """
-
-        self._evaluation_interval = evaluation_interval
-
-    @property
-    def min_interval(self):
-        """Gets the min_interval of this V1TruncationStoppingPolicy.  # noqa: E501
-
-
-        :return: The min_interval of this V1TruncationStoppingPolicy.  # noqa: E501
-        :rtype: int
-        """
-        return self._min_interval
-
-    @min_interval.setter
-    def min_interval(self, min_interval):
-        """Sets the min_interval of this V1TruncationStoppingPolicy.
-
-
-        :param min_interval: The min_interval of this V1TruncationStoppingPolicy.  # noqa: E501
-        :type min_interval: int
-        """
-
-        self._min_interval = min_interval
-
-    @property
-    def min_samples(self):
-        """Gets the min_samples of this V1TruncationStoppingPolicy.  # noqa: E501
-
-
-        :return: The min_samples of this V1TruncationStoppingPolicy.  # noqa: E501
-        :rtype: int
-        """
-        return self._min_samples
-
-    @min_samples.setter
-    def min_samples(self, min_samples):
-        """Sets the min_samples of this V1TruncationStoppingPolicy.
-
-
-        :param min_samples: The min_samples of this V1TruncationStoppingPolicy.  # noqa: E501
-        :type min_samples: int
-        """
-
-        self._min_samples = min_samples
-
-    @property
-    def include_succeeded(self):
-        """Gets the include_succeeded of this V1TruncationStoppingPolicy.  # noqa: E501
-
-
-        :return: The include_succeeded of this V1TruncationStoppingPolicy.  # noqa: E501
-        :rtype: bool
-        """
-        return self._include_succeeded
-
-    @include_succeeded.setter
-    def include_succeeded(self, include_succeeded):
-        """Sets the include_succeeded of this V1TruncationStoppingPolicy.
-
-
-        :param include_succeeded: The include_succeeded of this V1TruncationStoppingPolicy.  # noqa: E501
-        :type include_succeeded: bool
-        """
-
-        self._include_succeeded = include_succeeded
-
-    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, V1TruncationStoppingPolicy):
-            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, V1TruncationStoppingPolicy):
-            return True
+    kind: Optional[StrictStr] = 'truncation'
+    percent: Optional[StrictInt] = Field(None, description="The percentage of runs to stop, at each evaluation interval. e.g. 1 - 99.")
+    evaluation_interval: Optional[StrictInt] = Field(None, alias="evaluationInterval", description="Interval/Frequency for applying the policy.")
+    min_interval: Optional[StrictInt] = Field(None, alias="minInterval")
+    min_samples: Optional[StrictInt] = Field(None, alias="minSamples")
+    include_succeeded: Optional[StrictBool] = Field(None, alias="includeSucceeded")
+    __properties = ["kind", "percent", "evaluationInterval", "minInterval", "minSamples", "includeSucceeded"]
+
+    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) -> V1TruncationStoppingPolicy:
+        """Create an instance of V1TruncationStoppingPolicy 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) -> V1TruncationStoppingPolicy:
+        """Create an instance of V1TruncationStoppingPolicy from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1TruncationStoppingPolicy.parse_obj(obj)
+
+        _obj = V1TruncationStoppingPolicy.parse_obj({
+            "kind": obj.get("kind") if obj.get("kind") is not None else 'truncation',
+            "percent": obj.get("percent"),
+            "evaluation_interval": obj.get("evaluationInterval"),
+            "min_interval": obj.get("minInterval"),
+            "min_samples": obj.get("minSamples"),
+            "include_succeeded": obj.get("includeSucceeded")
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_tuner.py

@@ -1,226 +1,87 @@
-#!/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 V1Tuner(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
+from typing import Dict, List, Optional
+from pydantic import BaseModel, Field, StrictStr, conlist
+from polyaxon_sdk.models.v1_param import V1Param
 
+class V1Tuner(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.
+    V1Tuner
     """
-    openapi_types = {
-        'hub_ref': 'str',
-        'queue': 'str',
-        'presets': 'list[str]',
-        'params': 'dict(str, V1Param)'
-    }
-
-    attribute_map = {
-        'hub_ref': 'hubRef',
-        'queue': 'queue',
-        'presets': 'presets',
-        'params': 'params'
-    }
-
-    def __init__(self, hub_ref=None, queue=None, presets=None, params=None, local_vars_configuration=None):  # noqa: E501
-        """V1Tuner - 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._hub_ref = None
-        self._queue = None
-        self._presets = None
-        self._params = None
-        self.discriminator = None
-
-        if hub_ref is not None:
-            self.hub_ref = hub_ref
-        if queue is not None:
-            self.queue = queue
-        if presets is not None:
-            self.presets = presets
-        if params is not None:
-            self.params = params
-
-    @property
-    def hub_ref(self):
-        """Gets the hub_ref of this V1Tuner.  # noqa: E501
-
-
-        :return: The hub_ref of this V1Tuner.  # noqa: E501
-        :rtype: str
-        """
-        return self._hub_ref
-
-    @hub_ref.setter
-    def hub_ref(self, hub_ref):
-        """Sets the hub_ref of this V1Tuner.
-
-
-        :param hub_ref: The hub_ref of this V1Tuner.  # noqa: E501
-        :type hub_ref: str
-        """
-
-        self._hub_ref = hub_ref
-
-    @property
-    def queue(self):
-        """Gets the queue of this V1Tuner.  # noqa: E501
-
-
-        :return: The queue of this V1Tuner.  # noqa: E501
-        :rtype: str
-        """
-        return self._queue
-
-    @queue.setter
-    def queue(self, queue):
-        """Sets the queue of this V1Tuner.
-
-
-        :param queue: The queue of this V1Tuner.  # noqa: E501
-        :type queue: str
-        """
-
-        self._queue = queue
-
-    @property
-    def presets(self):
-        """Gets the presets of this V1Tuner.  # noqa: E501
-
-
-        :return: The presets of this V1Tuner.  # noqa: E501
-        :rtype: list[str]
-        """
-        return self._presets
-
-    @presets.setter
-    def presets(self, presets):
-        """Sets the presets of this V1Tuner.
-
-
-        :param presets: The presets of this V1Tuner.  # noqa: E501
-        :type presets: list[str]
-        """
-
-        self._presets = presets
-
-    @property
-    def params(self):
-        """Gets the params of this V1Tuner.  # noqa: E501
-
-
-        :return: The params of this V1Tuner.  # noqa: E501
-        :rtype: dict(str, V1Param)
-        """
-        return self._params
-
-    @params.setter
-    def params(self, params):
-        """Sets the params of this V1Tuner.
-
-
-        :param params: The params of this V1Tuner.  # noqa: E501
-        :type params: dict(str, V1Param)
-        """
-
-        self._params = params
-
-    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, V1Tuner):
-            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, V1Tuner):
-            return True
+    hub_ref: Optional[StrictStr] = Field(None, alias="hubRef")
+    presets: Optional[conlist(StrictStr)] = None
+    queue: Optional[StrictStr] = None
+    namespace: Optional[StrictStr] = None
+    params: Optional[Dict[str, V1Param]] = None
+    __properties = ["hubRef", "presets", "queue", "namespace", "params"]
+
+    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) -> V1Tuner:
+        """Create an instance of V1Tuner 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 value in params (dict)
+        _field_dict = {}
+        if self.params:
+            for _key in self.params:
+                if self.params[_key]:
+                    _field_dict[_key] = self.params[_key].to_dict()
+            _dict['params'] = _field_dict
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1Tuner:
+        """Create an instance of V1Tuner from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1Tuner.parse_obj(obj)
+
+        _obj = V1Tuner.parse_obj({
+            "hub_ref": obj.get("hubRef"),
+            "presets": obj.get("presets"),
+            "queue": obj.get("queue"),
+            "namespace": obj.get("namespace"),
+            "params": dict((_k, V1Param.from_dict(_v)) for _k, _v in obj.get("params").items()) if obj.get("params") is not None else None
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()