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

polyaxon_sdk/models/v1_bayes.py

@@ -1,382 +1,97 @@
-#!/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 Any, Dict, List, Optional
+from pydantic import BaseModel, Field, StrictInt, StrictStr, conlist
+from polyaxon_sdk.models.v1_optimization_metric import V1OptimizationMetric
+from polyaxon_sdk.models.v1_tuner import V1Tuner
 
-class V1Bayes(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
-
-    Do not edit the class manually.
-    """
-
+class V1Bayes(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.
+    V1Bayes
     """
-    openapi_types = {
-        'kind': 'str',
-        'params': 'dict(str, object)',
-        'num_initial_runs': 'int',
-        'max_iterations': 'int',
-        'utility_function': 'object',
-        'metric': 'V1OptimizationMetric',
-        'seed': 'int',
-        'concurrency': 'int',
-        'tuner': 'V1Tuner',
-        'early_stopping': 'list[object]'
-    }
-
-    attribute_map = {
-        'kind': 'kind',
-        'params': 'params',
-        'num_initial_runs': 'numInitialRuns',
-        'max_iterations': 'maxIterations',
-        'utility_function': 'utilityFunction',
-        'metric': 'metric',
-        'seed': 'seed',
-        'concurrency': 'concurrency',
-        'tuner': 'tuner',
-        'early_stopping': 'earlyStopping'
-    }
-
-    def __init__(self, kind='bayes', params=None, num_initial_runs=None, max_iterations=None, utility_function=None, metric=None, seed=None, concurrency=None, tuner=None, early_stopping=None, local_vars_configuration=None):  # noqa: E501
-        """V1Bayes - 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._params = None
-        self._num_initial_runs = None
-        self._max_iterations = None
-        self._utility_function = None
-        self._metric = None
-        self._seed = None
-        self._concurrency = None
-        self._tuner = None
-        self._early_stopping = None
-        self.discriminator = None
-
-        if kind is not None:
-            self.kind = kind
-        if params is not None:
-            self.params = params
-        if num_initial_runs is not None:
-            self.num_initial_runs = num_initial_runs
-        if max_iterations is not None:
-            self.max_iterations = max_iterations
-        if utility_function is not None:
-            self.utility_function = utility_function
-        if metric is not None:
-            self.metric = metric
-        if seed is not None:
-            self.seed = seed
-        if concurrency is not None:
-            self.concurrency = concurrency
-        if tuner is not None:
-            self.tuner = tuner
-        if early_stopping is not None:
-            self.early_stopping = early_stopping
-
-    @property
-    def kind(self):
-        """Gets the kind of this V1Bayes.  # noqa: E501
-
-
-        :return: The kind of this V1Bayes.  # noqa: E501
-        :rtype: str
-        """
-        return self._kind
-
-    @kind.setter
-    def kind(self, kind):
-        """Sets the kind of this V1Bayes.
-
-
-        :param kind: The kind of this V1Bayes.  # noqa: E501
-        :type kind: str
-        """
-
-        self._kind = kind
-
-    @property
-    def params(self):
-        """Gets the params of this V1Bayes.  # noqa: E501
-
-
-        :return: The params of this V1Bayes.  # noqa: E501
-        :rtype: dict(str, object)
-        """
-        return self._params
-
-    @params.setter
-    def params(self, params):
-        """Sets the params of this V1Bayes.
-
-
-        :param params: The params of this V1Bayes.  # noqa: E501
-        :type params: dict(str, object)
-        """
-
-        self._params = params
-
-    @property
-    def num_initial_runs(self):
-        """Gets the num_initial_runs of this V1Bayes.  # noqa: E501
-
-
-        :return: The num_initial_runs of this V1Bayes.  # noqa: E501
-        :rtype: int
-        """
-        return self._num_initial_runs
-
-    @num_initial_runs.setter
-    def num_initial_runs(self, num_initial_runs):
-        """Sets the num_initial_runs of this V1Bayes.
-
-
-        :param num_initial_runs: The num_initial_runs of this V1Bayes.  # noqa: E501
-        :type num_initial_runs: int
-        """
-
-        self._num_initial_runs = num_initial_runs
-
-    @property
-    def max_iterations(self):
-        """Gets the max_iterations of this V1Bayes.  # noqa: E501
-
-
-        :return: The max_iterations of this V1Bayes.  # noqa: E501
-        :rtype: int
-        """
-        return self._max_iterations
-
-    @max_iterations.setter
-    def max_iterations(self, max_iterations):
-        """Sets the max_iterations of this V1Bayes.
-
-
-        :param max_iterations: The max_iterations of this V1Bayes.  # noqa: E501
-        :type max_iterations: int
-        """
-
-        self._max_iterations = max_iterations
-
-    @property
-    def utility_function(self):
-        """Gets the utility_function of this V1Bayes.  # noqa: E501
-
-
-        :return: The utility_function of this V1Bayes.  # noqa: E501
-        :rtype: object
-        """
-        return self._utility_function
-
-    @utility_function.setter
-    def utility_function(self, utility_function):
-        """Sets the utility_function of this V1Bayes.
-
-
-        :param utility_function: The utility_function of this V1Bayes.  # noqa: E501
-        :type utility_function: object
-        """
-
-        self._utility_function = utility_function
-
-    @property
-    def metric(self):
-        """Gets the metric of this V1Bayes.  # noqa: E501
-
-
-        :return: The metric of this V1Bayes.  # noqa: E501
-        :rtype: V1OptimizationMetric
-        """
-        return self._metric
-
-    @metric.setter
-    def metric(self, metric):
-        """Sets the metric of this V1Bayes.
-
-
-        :param metric: The metric of this V1Bayes.  # noqa: E501
-        :type metric: V1OptimizationMetric
-        """
-
-        self._metric = metric
-
-    @property
-    def seed(self):
-        """Gets the seed of this V1Bayes.  # noqa: E501
-
-
-        :return: The seed of this V1Bayes.  # noqa: E501
-        :rtype: int
-        """
-        return self._seed
-
-    @seed.setter
-    def seed(self, seed):
-        """Sets the seed of this V1Bayes.
-
-
-        :param seed: The seed of this V1Bayes.  # noqa: E501
-        :type seed: int
-        """
-
-        self._seed = seed
-
-    @property
-    def concurrency(self):
-        """Gets the concurrency of this V1Bayes.  # noqa: E501
-
-
-        :return: The concurrency of this V1Bayes.  # noqa: E501
-        :rtype: int
-        """
-        return self._concurrency
-
-    @concurrency.setter
-    def concurrency(self, concurrency):
-        """Sets the concurrency of this V1Bayes.
-
-
-        :param concurrency: The concurrency of this V1Bayes.  # noqa: E501
-        :type concurrency: int
-        """
-
-        self._concurrency = concurrency
-
-    @property
-    def tuner(self):
-        """Gets the tuner of this V1Bayes.  # noqa: E501
-
-
-        :return: The tuner of this V1Bayes.  # noqa: E501
-        :rtype: V1Tuner
-        """
-        return self._tuner
-
-    @tuner.setter
-    def tuner(self, tuner):
-        """Sets the tuner of this V1Bayes.
-
-
-        :param tuner: The tuner of this V1Bayes.  # noqa: E501
-        :type tuner: V1Tuner
-        """
-
-        self._tuner = tuner
-
-    @property
-    def early_stopping(self):
-        """Gets the early_stopping of this V1Bayes.  # noqa: E501
-
-
-        :return: The early_stopping of this V1Bayes.  # noqa: E501
-        :rtype: list[object]
-        """
-        return self._early_stopping
-
-    @early_stopping.setter
-    def early_stopping(self, early_stopping):
-        """Sets the early_stopping of this V1Bayes.
-
-
-        :param early_stopping: The early_stopping of this V1Bayes.  # noqa: E501
-        :type early_stopping: list[object]
-        """
-
-        self._early_stopping = early_stopping
-
-    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, V1Bayes):
-            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, V1Bayes):
-            return True
+    kind: Optional[StrictStr] = 'bayes'
+    params: Optional[Dict[str, Dict[str, Any]]] = None
+    num_initial_runs: Optional[StrictInt] = Field(None, alias="numInitialRuns")
+    max_iterations: Optional[StrictInt] = Field(None, alias="maxIterations")
+    utility_function: Optional[Dict[str, Any]] = Field(None, alias="utilityFunction")
+    metric: Optional[V1OptimizationMetric] = None
+    seed: Optional[StrictInt] = None
+    concurrency: Optional[StrictInt] = None
+    tuner: Optional[V1Tuner] = None
+    early_stopping: Optional[conlist(Dict[str, Any])] = Field(None, alias="earlyStopping")
+    __properties = ["kind", "params", "numInitialRuns", "maxIterations", "utilityFunction", "metric", "seed", "concurrency", "tuner", "earlyStopping"]
+
+    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) -> V1Bayes:
+        """Create an instance of V1Bayes 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 metric
+        if self.metric:
+            _dict['metric'] = self.metric.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of tuner
+        if self.tuner:
+            _dict['tuner'] = self.tuner.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1Bayes:
+        """Create an instance of V1Bayes from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1Bayes.parse_obj(obj)
+
+        _obj = V1Bayes.parse_obj({
+            "kind": obj.get("kind") if obj.get("kind") is not None else 'bayes',
+            "params": obj.get("params"),
+            "num_initial_runs": obj.get("numInitialRuns"),
+            "max_iterations": obj.get("maxIterations"),
+            "utility_function": obj.get("utilityFunction"),
+            "metric": V1OptimizationMetric.from_dict(obj.get("metric")) if obj.get("metric") is not None else None,
+            "seed": obj.get("seed"),
+            "concurrency": obj.get("concurrency"),
+            "tuner": V1Tuner.from_dict(obj.get("tuner")) if obj.get("tuner") is not None else None,
+            "early_stopping": obj.get("earlyStopping")
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_bucket_connection.py

@@ -1,148 +1,71 @@
-#!/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 V1BucketConnection(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, StrictStr
 
+class V1BucketConnection(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.
+    V1BucketConnection
     """
-    openapi_types = {
-        'bucket': 'str'
-    }
-
-    attribute_map = {
-        'bucket': 'bucket'
-    }
-
-    def __init__(self, bucket=None, local_vars_configuration=None):  # noqa: E501
-        """V1BucketConnection - 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._bucket = None
-        self.discriminator = None
-
-        if bucket is not None:
-            self.bucket = bucket
-
-    @property
-    def bucket(self):
-        """Gets the bucket of this V1BucketConnection.  # noqa: E501
-
-
-        :return: The bucket of this V1BucketConnection.  # noqa: E501
-        :rtype: str
-        """
-        return self._bucket
-
-    @bucket.setter
-    def bucket(self, bucket):
-        """Sets the bucket of this V1BucketConnection.
-
-
-        :param bucket: The bucket of this V1BucketConnection.  # noqa: E501
-        :type bucket: str
-        """
-
-        self._bucket = bucket
-
-    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, V1BucketConnection):
-            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, V1BucketConnection):
-            return True
-
-        return self.to_dict() != other.to_dict()
+    bucket: Optional[StrictStr] = None
+    __properties = ["bucket"]
+
+    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) -> V1BucketConnection:
+        """Create an instance of V1BucketConnection 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) -> V1BucketConnection:
+        """Create an instance of V1BucketConnection from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1BucketConnection.parse_obj(obj)
+
+        _obj = V1BucketConnection.parse_obj({
+            "bucket": obj.get("bucket")
+        })
+        return _obj
+