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

polyaxon_sdk/models/v1_xg_boost_job.py

@@ -1,252 +1,91 @@
-#!/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 V1XGBoostJob(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, StrictStr
+from polyaxon_sdk.models.v1_clean_pod_policy import V1CleanPodPolicy
+from polyaxon_sdk.models.v1_kf_replica import V1KFReplica
+from polyaxon_sdk.models.v1_scheduling_policy import V1SchedulingPolicy
 
+class V1XGBoostJob(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.
+    V1XGBoostJob
     """
-    openapi_types = {
-        'kind': 'str',
-        'clean_pod_policy': 'V1CleanPodPolicy',
-        'scheduling_policy': 'V1SchedulingPolicy',
-        'master': 'V1KFReplica',
-        'worker': 'V1KFReplica'
-    }
-
-    attribute_map = {
-        'kind': 'kind',
-        'clean_pod_policy': 'cleanPodPolicy',
-        'scheduling_policy': 'schedulingPolicy',
-        'master': 'master',
-        'worker': 'worker'
-    }
-
-    def __init__(self, kind='xgboost_job', clean_pod_policy=None, scheduling_policy=None, master=None, worker=None, local_vars_configuration=None):  # noqa: E501
-        """V1XGBoostJob - 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._clean_pod_policy = None
-        self._scheduling_policy = None
-        self._master = None
-        self._worker = None
-        self.discriminator = None
-
-        if kind is not None:
-            self.kind = kind
-        if clean_pod_policy is not None:
-            self.clean_pod_policy = clean_pod_policy
-        if scheduling_policy is not None:
-            self.scheduling_policy = scheduling_policy
-        if master is not None:
-            self.master = master
-        if worker is not None:
-            self.worker = worker
-
-    @property
-    def kind(self):
-        """Gets the kind of this V1XGBoostJob.  # noqa: E501
-
-
-        :return: The kind of this V1XGBoostJob.  # noqa: E501
-        :rtype: str
-        """
-        return self._kind
-
-    @kind.setter
-    def kind(self, kind):
-        """Sets the kind of this V1XGBoostJob.
-
-
-        :param kind: The kind of this V1XGBoostJob.  # noqa: E501
-        :type kind: str
-        """
-
-        self._kind = kind
-
-    @property
-    def clean_pod_policy(self):
-        """Gets the clean_pod_policy of this V1XGBoostJob.  # noqa: E501
-
-
-        :return: The clean_pod_policy of this V1XGBoostJob.  # noqa: E501
-        :rtype: V1CleanPodPolicy
-        """
-        return self._clean_pod_policy
-
-    @clean_pod_policy.setter
-    def clean_pod_policy(self, clean_pod_policy):
-        """Sets the clean_pod_policy of this V1XGBoostJob.
-
-
-        :param clean_pod_policy: The clean_pod_policy of this V1XGBoostJob.  # noqa: E501
-        :type clean_pod_policy: V1CleanPodPolicy
-        """
-
-        self._clean_pod_policy = clean_pod_policy
-
-    @property
-    def scheduling_policy(self):
-        """Gets the scheduling_policy of this V1XGBoostJob.  # noqa: E501
-
-
-        :return: The scheduling_policy of this V1XGBoostJob.  # noqa: E501
-        :rtype: V1SchedulingPolicy
-        """
-        return self._scheduling_policy
-
-    @scheduling_policy.setter
-    def scheduling_policy(self, scheduling_policy):
-        """Sets the scheduling_policy of this V1XGBoostJob.
-
-
-        :param scheduling_policy: The scheduling_policy of this V1XGBoostJob.  # noqa: E501
-        :type scheduling_policy: V1SchedulingPolicy
-        """
-
-        self._scheduling_policy = scheduling_policy
-
-    @property
-    def master(self):
-        """Gets the master of this V1XGBoostJob.  # noqa: E501
-
-
-        :return: The master of this V1XGBoostJob.  # noqa: E501
-        :rtype: V1KFReplica
-        """
-        return self._master
-
-    @master.setter
-    def master(self, master):
-        """Sets the master of this V1XGBoostJob.
-
-
-        :param master: The master of this V1XGBoostJob.  # noqa: E501
-        :type master: V1KFReplica
-        """
-
-        self._master = master
-
-    @property
-    def worker(self):
-        """Gets the worker of this V1XGBoostJob.  # noqa: E501
-
-
-        :return: The worker of this V1XGBoostJob.  # noqa: E501
-        :rtype: V1KFReplica
-        """
-        return self._worker
-
-    @worker.setter
-    def worker(self, worker):
-        """Sets the worker of this V1XGBoostJob.
-
-
-        :param worker: The worker of this V1XGBoostJob.  # noqa: E501
-        :type worker: V1KFReplica
-        """
-
-        self._worker = worker
-
-    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, V1XGBoostJob):
-            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, V1XGBoostJob):
-            return True
+    kind: Optional[StrictStr] = 'xgboostjob'
+    clean_pod_policy: Optional[V1CleanPodPolicy] = Field(None, alias="cleanPodPolicy")
+    scheduling_policy: Optional[V1SchedulingPolicy] = Field(None, alias="schedulingPolicy")
+    master: Optional[V1KFReplica] = None
+    worker: Optional[V1KFReplica] = None
+    __properties = ["kind", "cleanPodPolicy", "schedulingPolicy", "master", "worker"]
+
+    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) -> V1XGBoostJob:
+        """Create an instance of V1XGBoostJob 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 scheduling_policy
+        if self.scheduling_policy:
+            _dict['schedulingPolicy'] = self.scheduling_policy.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of master
+        if self.master:
+            _dict['master'] = self.master.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of worker
+        if self.worker:
+            _dict['worker'] = self.worker.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> V1XGBoostJob:
+        """Create an instance of V1XGBoostJob from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1XGBoostJob.parse_obj(obj)
+
+        _obj = V1XGBoostJob.parse_obj({
+            "kind": obj.get("kind") if obj.get("kind") is not None else 'xgboostjob',
+            "clean_pod_policy": obj.get("cleanPodPolicy"),
+            "scheduling_policy": V1SchedulingPolicy.from_dict(obj.get("schedulingPolicy")) if obj.get("schedulingPolicy") is not None else None,
+            "master": V1KFReplica.from_dict(obj.get("master")) if obj.get("master") is not None else None,
+            "worker": V1KFReplica.from_dict(obj.get("worker")) if obj.get("worker") is not None else None
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/rest.py

@@ -1,33 +1,17 @@
-#!/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.
+"""
 
-from __future__ import absolute_import
 
 import io
 import json
@@ -35,9 +19,7 @@ import logging
 import re
 import ssl
 
-# python 2 and python 3 compatibility library
-import six
-from six.moves.urllib.parse import urlencode
+from urllib.parse import urlencode, quote_plus
 import urllib3
 
 from polyaxon_sdk.exceptions import ApiException, UnauthorizedException, ForbiddenException, NotFoundException, ServiceException, ApiValueError
@@ -56,11 +38,11 @@ class RESTResponse(io.IOBase):
 
     def getheaders(self):
         """Returns a dictionary of the response headers."""
-        return self.urllib3_response.getheaders()
+        return self.urllib3_response.headers
 
     def getheader(self, name, default=None):
         """Returns a given response header."""
-        return self.urllib3_response.getheader(name, default)
+        return self.urllib3_response.headers.get(name, default)
 
 
 class RESTClientObject(object):
@@ -150,25 +132,25 @@ class RESTClientObject(object):
 
         post_params = post_params or {}
         headers = headers or {}
+        # url already contains the URL query string
+        # so reset query_params to empty dict
+        query_params = {}
 
         timeout = None
         if _request_timeout:
-            if isinstance(_request_timeout, six.integer_types + (float, )):  # noqa: E501,F821
+            if isinstance(_request_timeout, (int,float)):  # noqa: E501,F821
                 timeout = urllib3.Timeout(total=_request_timeout)
             elif (isinstance(_request_timeout, tuple) and
                   len(_request_timeout) == 2):
                 timeout = urllib3.Timeout(
                     connect=_request_timeout[0], read=_request_timeout[1])
 
-        if 'Content-Type' not in headers:
-            headers['Content-Type'] = 'application/json'
-
         try:
             # For `POST`, `PUT`, `PATCH`, `OPTIONS`, `DELETE`
             if method in ['POST', 'PUT', 'PATCH', 'OPTIONS', 'DELETE']:
-                if query_params:
-                    url += '?' + urlencode(query_params)
-                if re.search('json', headers['Content-Type'], re.IGNORECASE):
+
+                # no content type provided or payload is json
+                if not headers.get('Content-Type') or re.search('json', headers['Content-Type'], re.IGNORECASE):
                     request_body = None
                     if body is not None:
                         request_body = json.dumps(body)
@@ -218,7 +200,7 @@ class RESTClientObject(object):
             # For `GET`, `HEAD`
             else:
                 r = self.pool_manager.request(method, url,
-                                              fields=query_params,
+                                              fields={},
                                               preload_content=_preload_content,
                                               timeout=timeout,
                                               headers=headers)
@@ -249,7 +231,7 @@ class RESTClientObject(object):
 
         return r
 
-    def GET(self, url, headers=None, query_params=None, _preload_content=True,
+    def get_request(self, url, headers=None, query_params=None, _preload_content=True,
             _request_timeout=None):
         return self.request("GET", url,
                             headers=headers,
@@ -257,7 +239,7 @@ class RESTClientObject(object):
                             _request_timeout=_request_timeout,
                             query_params=query_params)
 
-    def HEAD(self, url, headers=None, query_params=None, _preload_content=True,
+    def head_request(self, url, headers=None, query_params=None, _preload_content=True,
              _request_timeout=None):
         return self.request("HEAD", url,
                             headers=headers,
@@ -265,7 +247,7 @@ class RESTClientObject(object):
                             _request_timeout=_request_timeout,
                             query_params=query_params)
 
-    def OPTIONS(self, url, headers=None, query_params=None, post_params=None,
+    def options_request(self, url, headers=None, query_params=None, post_params=None,
                 body=None, _preload_content=True, _request_timeout=None):
         return self.request("OPTIONS", url,
                             headers=headers,
@@ -275,7 +257,7 @@ class RESTClientObject(object):
                             _request_timeout=_request_timeout,
                             body=body)
 
-    def DELETE(self, url, headers=None, query_params=None, body=None,
+    def delete_request(self, url, headers=None, query_params=None, body=None,
                _preload_content=True, _request_timeout=None):
         return self.request("DELETE", url,
                             headers=headers,
@@ -284,7 +266,7 @@ class RESTClientObject(object):
                             _request_timeout=_request_timeout,
                             body=body)
 
-    def POST(self, url, headers=None, query_params=None, post_params=None,
+    def post_request(self, url, headers=None, query_params=None, post_params=None,
              body=None, _preload_content=True, _request_timeout=None):
         return self.request("POST", url,
                             headers=headers,
@@ -294,7 +276,7 @@ class RESTClientObject(object):
                             _request_timeout=_request_timeout,
                             body=body)
 
-    def PUT(self, url, headers=None, query_params=None, post_params=None,
+    def put_request(self, url, headers=None, query_params=None, post_params=None,
             body=None, _preload_content=True, _request_timeout=None):
         return self.request("PUT", url,
                             headers=headers,
@@ -304,7 +286,7 @@ class RESTClientObject(object):
                             _request_timeout=_request_timeout,
                             body=body)
 
-    def PATCH(self, url, headers=None, query_params=None, post_params=None,
+    def patch_request(self, url, headers=None, query_params=None, post_params=None,
               body=None, _preload_content=True, _request_timeout=None):
         return self.request("PATCH", url,
                             headers=headers,

{polyaxon_sdk-1.18.0rc7.dist-info → polyaxon_sdk-2.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: polyaxon-sdk
-Version: 1.18.0rc7
+Version: 2.1.6
 Summary: Polyaxon SDKs and REST API specification.
 Home-page: UNKNOWN
 Author: Polyaxon sdk
@@ -8,10 +8,12 @@ Author-email: contact@polyaxon.com
 License: UNKNOWN
 Keywords: OpenAPI,OpenAPI-Generator,Polyaxon SDKs and REST API specification.
 Platform: UNKNOWN
-Requires-Dist: urllib3 (>=1.25.3)
-Requires-Dist: six (>=1.10)
+Description-Content-Type: text/markdown
+Requires-Dist: urllib3 >=1.25.3
 Requires-Dist: python-dateutil
+Requires-Dist: pydantic
+Requires-Dist: aenum
 
-    Polyaxon SDKs and REST API specification.  # noqa: E501
+         # noqa: E501