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

polyaxon_sdk/models/v1_matrix.py

@@ -1,304 +1,111 @@
-#!/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
+import json
 
-from polyaxon_sdk.configuration import Configuration
 
+from typing import Optional
+from pydantic import BaseModel
+from polyaxon_sdk.models.v1_bayes import V1Bayes
+from polyaxon_sdk.models.v1_grid_search import V1GridSearch
+from polyaxon_sdk.models.v1_hyperband import V1Hyperband
+from polyaxon_sdk.models.v1_hyperopt import V1Hyperopt
+from polyaxon_sdk.models.v1_iterative import V1Iterative
+from polyaxon_sdk.models.v1_mapping import V1Mapping
+from polyaxon_sdk.models.v1_random_search import V1RandomSearch
 
-class V1Matrix(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
+class V1Matrix(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.
+    V1Matrix
     """
-    openapi_types = {
-        'random': 'V1RandomSearch',
-        'grid': 'V1GridSearch',
-        'hyperband': 'V1Hyperband',
-        'bayes': 'V1Bayes',
-        'hyperopt': 'V1Hyperopt',
-        'iterative': 'V1Iterative',
-        'mapping': 'V1Mapping'
-    }
-
-    attribute_map = {
-        'random': 'random',
-        'grid': 'grid',
-        'hyperband': 'hyperband',
-        'bayes': 'bayes',
-        'hyperopt': 'hyperopt',
-        'iterative': 'iterative',
-        'mapping': 'mapping'
-    }
-
-    def __init__(self, random=None, grid=None, hyperband=None, bayes=None, hyperopt=None, iterative=None, mapping=None, local_vars_configuration=None):  # noqa: E501
-        """V1Matrix - 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._random = None
-        self._grid = None
-        self._hyperband = None
-        self._bayes = None
-        self._hyperopt = None
-        self._iterative = None
-        self._mapping = None
-        self.discriminator = None
-
-        if random is not None:
-            self.random = random
-        if grid is not None:
-            self.grid = grid
-        if hyperband is not None:
-            self.hyperband = hyperband
-        if bayes is not None:
-            self.bayes = bayes
-        if hyperopt is not None:
-            self.hyperopt = hyperopt
-        if iterative is not None:
-            self.iterative = iterative
-        if mapping is not None:
-            self.mapping = mapping
-
-    @property
-    def random(self):
-        """Gets the random of this V1Matrix.  # noqa: E501
-
-
-        :return: The random of this V1Matrix.  # noqa: E501
-        :rtype: V1RandomSearch
-        """
-        return self._random
-
-    @random.setter
-    def random(self, random):
-        """Sets the random of this V1Matrix.
-
-
-        :param random: The random of this V1Matrix.  # noqa: E501
-        :type random: V1RandomSearch
-        """
-
-        self._random = random
-
-    @property
-    def grid(self):
-        """Gets the grid of this V1Matrix.  # noqa: E501
-
-
-        :return: The grid of this V1Matrix.  # noqa: E501
-        :rtype: V1GridSearch
-        """
-        return self._grid
-
-    @grid.setter
-    def grid(self, grid):
-        """Sets the grid of this V1Matrix.
-
-
-        :param grid: The grid of this V1Matrix.  # noqa: E501
-        :type grid: V1GridSearch
-        """
-
-        self._grid = grid
-
-    @property
-    def hyperband(self):
-        """Gets the hyperband of this V1Matrix.  # noqa: E501
-
-
-        :return: The hyperband of this V1Matrix.  # noqa: E501
-        :rtype: V1Hyperband
-        """
-        return self._hyperband
-
-    @hyperband.setter
-    def hyperband(self, hyperband):
-        """Sets the hyperband of this V1Matrix.
-
-
-        :param hyperband: The hyperband of this V1Matrix.  # noqa: E501
-        :type hyperband: V1Hyperband
-        """
-
-        self._hyperband = hyperband
-
-    @property
-    def bayes(self):
-        """Gets the bayes of this V1Matrix.  # noqa: E501
-
-
-        :return: The bayes of this V1Matrix.  # noqa: E501
-        :rtype: V1Bayes
-        """
-        return self._bayes
-
-    @bayes.setter
-    def bayes(self, bayes):
-        """Sets the bayes of this V1Matrix.
-
-
-        :param bayes: The bayes of this V1Matrix.  # noqa: E501
-        :type bayes: V1Bayes
-        """
-
-        self._bayes = bayes
-
-    @property
-    def hyperopt(self):
-        """Gets the hyperopt of this V1Matrix.  # noqa: E501
-
-
-        :return: The hyperopt of this V1Matrix.  # noqa: E501
-        :rtype: V1Hyperopt
-        """
-        return self._hyperopt
-
-    @hyperopt.setter
-    def hyperopt(self, hyperopt):
-        """Sets the hyperopt of this V1Matrix.
-
-
-        :param hyperopt: The hyperopt of this V1Matrix.  # noqa: E501
-        :type hyperopt: V1Hyperopt
-        """
-
-        self._hyperopt = hyperopt
-
-    @property
-    def iterative(self):
-        """Gets the iterative of this V1Matrix.  # noqa: E501
-
-
-        :return: The iterative of this V1Matrix.  # noqa: E501
-        :rtype: V1Iterative
-        """
-        return self._iterative
-
-    @iterative.setter
-    def iterative(self, iterative):
-        """Sets the iterative of this V1Matrix.
-
-
-        :param iterative: The iterative of this V1Matrix.  # noqa: E501
-        :type iterative: V1Iterative
-        """
-
-        self._iterative = iterative
-
-    @property
-    def mapping(self):
-        """Gets the mapping of this V1Matrix.  # noqa: E501
-
-
-        :return: The mapping of this V1Matrix.  # noqa: E501
-        :rtype: V1Mapping
-        """
-        return self._mapping
-
-    @mapping.setter
-    def mapping(self, mapping):
-        """Sets the mapping of this V1Matrix.
-
-
-        :param mapping: The mapping of this V1Matrix.  # noqa: E501
-        :type mapping: V1Mapping
-        """
-
-        self._mapping = mapping
-
-    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, V1Matrix):
-            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, V1Matrix):
-            return True
+    random: Optional[V1RandomSearch] = None
+    grid: Optional[V1GridSearch] = None
+    hyperband: Optional[V1Hyperband] = None
+    bayes: Optional[V1Bayes] = None
+    hyperopt: Optional[V1Hyperopt] = None
+    iterative: Optional[V1Iterative] = None
+    mapping: Optional[V1Mapping] = None
+    __properties = ["random", "grid", "hyperband", "bayes", "hyperopt", "iterative", "mapping"]
+
+    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) -> V1Matrix:
+        """Create an instance of V1Matrix 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 random
+        if self.random:
+            _dict['random'] = self.random.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of grid
+        if self.grid:
+            _dict['grid'] = self.grid.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of hyperband
+        if self.hyperband:
+            _dict['hyperband'] = self.hyperband.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of bayes
+        if self.bayes:
+            _dict['bayes'] = self.bayes.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of hyperopt
+        if self.hyperopt:
+            _dict['hyperopt'] = self.hyperopt.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of iterative
+        if self.iterative:
+            _dict['iterative'] = self.iterative.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of mapping
+        if self.mapping:
+            _dict['mapping'] = self.mapping.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1Matrix:
+        """Create an instance of V1Matrix from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1Matrix.parse_obj(obj)
+
+        _obj = V1Matrix.parse_obj({
+            "random": V1RandomSearch.from_dict(obj.get("random")) if obj.get("random") is not None else None,
+            "grid": V1GridSearch.from_dict(obj.get("grid")) if obj.get("grid") is not None else None,
+            "hyperband": V1Hyperband.from_dict(obj.get("hyperband")) if obj.get("hyperband") is not None else None,
+            "bayes": V1Bayes.from_dict(obj.get("bayes")) if obj.get("bayes") is not None else None,
+            "hyperopt": V1Hyperopt.from_dict(obj.get("hyperopt")) if obj.get("hyperopt") is not None else None,
+            "iterative": V1Iterative.from_dict(obj.get("iterative")) if obj.get("iterative") is not None else None,
+            "mapping": V1Mapping.from_dict(obj.get("mapping")) if obj.get("mapping") is not None else None
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_matrix_kind.py

@@ -1,133 +1,40 @@
-#!/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 V1MatrixKind(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
 
-    Do not edit the class manually.
-    """
 
+class V1MatrixKind(str, Enum):
     """
-    allowed enum values
+    V1MatrixKind
     """
-    RANDOM = "random"
-    GRID = "grid"
-    HYPERBAND = "hyperband"
-    BAYES = "bayes"
-    HYPEROPT = "hyperopt"
-    ITERATIVE = "iterative"
-    MAPPING = "mapping"
-
-    allowable_values = [RANDOM, GRID, HYPERBAND, BAYES, HYPEROPT, ITERATIVE, MAPPING]  # 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
-        """V1MatrixKind - 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, V1MatrixKind):
-            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, V1MatrixKind):
-            return True
+    RANDOM = 'random'
+    GRID = 'grid'
+    HYPERBAND = 'hyperband'
+    BAYES = 'bayes'
+    HYPEROPT = 'hyperopt'
+    ITERATIVE = 'iterative'
+    MAPPING = 'mapping'
 
-        return self.to_dict() != other.to_dict()