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

polyaxon_sdk/models/v1_project_version_kind.py

@@ -1,129 +1,36 @@
-#!/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 V1ProjectVersionKind(object):
-    """NOTE: This class is auto generated by OpenAPI Generator.
-    Ref: https://openapi-generator.tech
 
-    Do not edit the class manually.
-    """
 
+class V1ProjectVersionKind(str, Enum):
     """
-    allowed enum values
+    V1ProjectVersionKind
     """
-    COMPONENT = "component"
-    MODEL = "model"
-    ARTIFACT = "artifact"
-
-    allowable_values = [COMPONENT, MODEL, ARTIFACT]  # 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
-        """V1ProjectVersionKind - 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, V1ProjectVersionKind):
-            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, V1ProjectVersionKind):
-            return True
+    COMPONENT = 'component'
+    MODEL = 'model'
+    ARTIFACT = 'artifact'
 
-        return self.to_dict() != other.to_dict()

polyaxon_sdk/models/v1_pytorch_elastic_policy.py

@@ -0,0 +1,91 @@
+# coding: utf-8
+
+"""
+    Polyaxon SDKs and REST API specification.
+
+         # noqa: E501
+
+    The version of the OpenAPI document: 2.1.6
+    Contact: contact@polyaxon.com
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""
+
+
+from __future__ import annotations
+from inspect import getfullargspec
+import pprint
+import re  # noqa: F401
+import json
+
+
+from typing import Any, Dict, List, Optional
+from pydantic import BaseModel, Field, StrictBool, StrictInt, StrictStr, conlist
+
+class V1PytorchElasticPolicy(BaseModel):
+    """
+    V1PytorchElasticPolicy
+    """
+    min_replicas: Optional[StrictInt] = Field(None, alias="minReplicas", description="minReplicas is the lower limit for the number of replicas to which the training job can scale down.")
+    max_replicas: Optional[StrictInt] = Field(None, alias="maxReplicas", description="upper limit for the number of pods that can be set by the autoscaler; cannot be smaller than MinReplicas, defaults to null.")
+    rdvz_backend: Optional[StrictStr] = Field(None, alias="rdvzBackend")
+    rdvz_port: Optional[StrictInt] = Field(None, alias="rdvzPort")
+    rdvz_host: Optional[StrictStr] = Field(None, alias="rdvzHost")
+    rdvz_id: Optional[StrictStr] = Field(None, alias="rdvzId")
+    rdvz_conf: Optional[conlist(Dict[str, Any])] = Field(None, alias="rdvzConf", description="RDZVConf contains additional rendezvous configuration (<key1>=<value1>,<key2>=<value2>,...).")
+    standalone: Optional[StrictBool] = Field(None, description="Start a local standalone rendezvous backend that is represented by a C10d TCP store on port 29400. Useful when launching single-node, multi-worker job. If specified --rdzv_backend, --rdzv_endpoint, --rdzv_id are auto-assigned; any explicitly set values are ignored.")
+    n_proc_per_node: Optional[StrictInt] = Field(None, alias="nProcPerNode", description="Number of workers per node; supported values: [auto, cpu, gpu, int]. Deprecated: This API is deprecated in v1.7+ Use .spec.nprocPerNode instead.")
+    max_restarts: Optional[StrictInt] = Field(None, alias="maxRestarts")
+    metrics: Optional[conlist(Dict[str, Any])] = Field(None, alias="Metrics", description="Metrics contains the specifications which are used to calculate the desired replica count (the maximum replica count across all metrics will be used).  The desired replica count is calculated with multiplying the ratio between the target value and the current value by the current number of pods. Ergo, metrics used must decrease as the pod count is increased, and vice-versa.  See the individual metric source types for more information about how each type of metric must respond. If not set, the HPA will not be created.")
+    __properties = ["minReplicas", "maxReplicas", "rdvzBackend", "rdvzPort", "rdvzHost", "rdvzId", "rdvzConf", "standalone", "nProcPerNode", "maxRestarts", "Metrics"]
+
+    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) -> V1PytorchElasticPolicy:
+        """Create an instance of V1PytorchElasticPolicy 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) -> V1PytorchElasticPolicy:
+        """Create an instance of V1PytorchElasticPolicy from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1PytorchElasticPolicy.parse_obj(obj)
+
+        _obj = V1PytorchElasticPolicy.parse_obj({
+            "min_replicas": obj.get("minReplicas"),
+            "max_replicas": obj.get("maxReplicas"),
+            "rdvz_backend": obj.get("rdvzBackend"),
+            "rdvz_port": obj.get("rdvzPort"),
+            "rdvz_host": obj.get("rdvzHost"),
+            "rdvz_id": obj.get("rdvzId"),
+            "rdvz_conf": obj.get("rdvzConf"),
+            "standalone": obj.get("standalone"),
+            "n_proc_per_node": obj.get("nProcPerNode"),
+            "max_restarts": obj.get("maxRestarts"),
+            "metrics": obj.get("Metrics")
+        })
+        return _obj
+

polyaxon_sdk/models/v1_pytorch_job.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 V1PytorchJob(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_pytorch_elastic_policy import V1PytorchElasticPolicy
+from polyaxon_sdk.models.v1_scheduling_policy import V1SchedulingPolicy
 
+class V1PytorchJob(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.
+    V1PytorchJob
     """
-    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='pytorch_job', clean_pod_policy=None, scheduling_policy=None, master=None, worker=None, local_vars_configuration=None):  # noqa: E501
-        """V1PytorchJob - 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 V1PytorchJob.  # noqa: E501
-
-
-        :return: The kind of this V1PytorchJob.  # noqa: E501
-        :rtype: str
-        """
-        return self._kind
-
-    @kind.setter
-    def kind(self, kind):
-        """Sets the kind of this V1PytorchJob.
-
-
-        :param kind: The kind of this V1PytorchJob.  # noqa: E501
-        :type kind: str
-        """
-
-        self._kind = kind
-
-    @property
-    def clean_pod_policy(self):
-        """Gets the clean_pod_policy of this V1PytorchJob.  # noqa: E501
-
-
-        :return: The clean_pod_policy of this V1PytorchJob.  # 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 V1PytorchJob.
-
-
-        :param clean_pod_policy: The clean_pod_policy of this V1PytorchJob.  # 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 V1PytorchJob.  # noqa: E501
-
-
-        :return: The scheduling_policy of this V1PytorchJob.  # noqa: E501
-        :rtype: V1SchedulingPolicy
-        """
-        return self._scheduling_policy
-
-    @scheduling_policy.setter
-    def scheduling_policy(self, scheduling_policy):
-        """Sets the scheduling_policy of this V1PytorchJob.
-
-
-        :param scheduling_policy: The scheduling_policy of this V1PytorchJob.  # noqa: E501
-        :type scheduling_policy: V1SchedulingPolicy
-        """
-
-        self._scheduling_policy = scheduling_policy
-
-    @property
-    def master(self):
-        """Gets the master of this V1PytorchJob.  # noqa: E501
-
-
-        :return: The master of this V1PytorchJob.  # noqa: E501
-        :rtype: V1KFReplica
-        """
-        return self._master
-
-    @master.setter
-    def master(self, master):
-        """Sets the master of this V1PytorchJob.
-
-
-        :param master: The master of this V1PytorchJob.  # noqa: E501
-        :type master: V1KFReplica
-        """
-
-        self._master = master
-
-    @property
-    def worker(self):
-        """Gets the worker of this V1PytorchJob.  # noqa: E501
-
-
-        :return: The worker of this V1PytorchJob.  # noqa: E501
-        :rtype: V1KFReplica
-        """
-        return self._worker
-
-    @worker.setter
-    def worker(self, worker):
-        """Sets the worker of this V1PytorchJob.
-
-
-        :param worker: The worker of this V1PytorchJob.  # 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, V1PytorchJob):
-            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, V1PytorchJob):
-            return True
+    kind: Optional[StrictStr] = 'pytorchjob'
+    clean_pod_policy: Optional[V1CleanPodPolicy] = Field(None, alias="cleanPodPolicy")
+    scheduling_policy: Optional[V1SchedulingPolicy] = Field(None, alias="schedulingPolicy")
+    elastic_policy: Optional[V1PytorchElasticPolicy] = Field(None, alias="elasticPolicy")
+    master: Optional[V1KFReplica] = None
+    worker: Optional[V1KFReplica] = None
+    n_proc_per_node: Optional[StrictStr] = Field(None, alias="nProcPerNode")
+    __properties = ["kind", "cleanPodPolicy", "schedulingPolicy", "elasticPolicy", "master", "worker", "nProcPerNode"]
+
+    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) -> V1PytorchJob:
+        """Create an instance of V1PytorchJob 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 elastic_policy
+        if self.elastic_policy:
+            _dict['elasticPolicy'] = self.elastic_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) -> V1PytorchJob:
+        """Create an instance of V1PytorchJob from a dict"""
+        if obj is None:
+            return None
+
+        if type(obj) is not dict:
+            return V1PytorchJob.parse_obj(obj)
+
+        _obj = V1PytorchJob.parse_obj({
+            "kind": obj.get("kind") if obj.get("kind") is not None else 'pytorchjob',
+            "clean_pod_policy": obj.get("cleanPodPolicy"),
+            "scheduling_policy": V1SchedulingPolicy.from_dict(obj.get("schedulingPolicy")) if obj.get("schedulingPolicy") is not None else None,
+            "elastic_policy": V1PytorchElasticPolicy.from_dict(obj.get("elasticPolicy")) if obj.get("elasticPolicy") 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,
+            "n_proc_per_node": obj.get("nProcPerNode")
+        })
+        return _obj
 
-        return self.to_dict() != other.to_dict()