anyscale 0.24.86__py3-none-any.whl

anyscale/sdk/anyscale_client/models/production_job_config.py

@@ -0,0 +1,348 @@
+# coding: utf-8
+
+"""
+    Anyscale API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from anyscale_client.configuration import Configuration
+
+
+class ProductionJobConfig(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    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.
+    """
+    openapi_types = {
+        'entrypoint': 'str',
+        'ray_serve_config': 'object',
+        'runtime_env': 'RayRuntimeEnvConfig',
+        'build_id': 'str',
+        'compute_config_id': 'str',
+        'compute_config': 'CreateClusterComputeConfig',
+        'max_retries': 'int',
+        'timeout_s': 'int',
+        'runtime_env_config': 'RayRuntimeEnvConfig'
+    }
+
+    attribute_map = {
+        'entrypoint': 'entrypoint',
+        'ray_serve_config': 'ray_serve_config',
+        'runtime_env': 'runtime_env',
+        'build_id': 'build_id',
+        'compute_config_id': 'compute_config_id',
+        'compute_config': 'compute_config',
+        'max_retries': 'max_retries',
+        'timeout_s': 'timeout_s',
+        'runtime_env_config': 'runtime_env_config'
+    }
+
+    def __init__(self, entrypoint='', ray_serve_config=None, runtime_env=None, build_id=None, compute_config_id=None, compute_config=None, max_retries=5, timeout_s=None, runtime_env_config=None, local_vars_configuration=None):  # noqa: E501
+        """ProductionJobConfig - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+
+        self._entrypoint = None
+        self._ray_serve_config = None
+        self._runtime_env = None
+        self._build_id = None
+        self._compute_config_id = None
+        self._compute_config = None
+        self._max_retries = None
+        self._timeout_s = None
+        self._runtime_env_config = None
+        self.discriminator = None
+
+        if entrypoint is not None:
+            self.entrypoint = entrypoint
+        if ray_serve_config is not None:
+            self.ray_serve_config = ray_serve_config
+        if runtime_env is not None:
+            self.runtime_env = runtime_env
+        self.build_id = build_id
+        self.compute_config_id = compute_config_id
+        if compute_config is not None:
+            self.compute_config = compute_config
+        if max_retries is not None:
+            self.max_retries = max_retries
+        if timeout_s is not None:
+            self.timeout_s = timeout_s
+        if runtime_env_config is not None:
+            self.runtime_env_config = runtime_env_config
+
+    @property
+    def entrypoint(self):
+        """Gets the entrypoint of this ProductionJobConfig.  # noqa: E501
+
+        A script that will be run to start your job.This command will be run in the root directory of the specified runtime env. Eg. 'python script.py'  # noqa: E501
+
+        :return: The entrypoint of this ProductionJobConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._entrypoint
+
+    @entrypoint.setter
+    def entrypoint(self, entrypoint):
+        """Sets the entrypoint of this ProductionJobConfig.
+
+        A script that will be run to start your job.This command will be run in the root directory of the specified runtime env. Eg. 'python script.py'  # noqa: E501
+
+        :param entrypoint: The entrypoint of this ProductionJobConfig.  # noqa: E501
+        :type: str
+        """
+
+        self._entrypoint = entrypoint
+
+    @property
+    def ray_serve_config(self):
+        """Gets the ray_serve_config of this ProductionJobConfig.  # noqa: E501
+
+        The Ray Serve config to use for this Production service. This config defines your Ray Serve application, and will be passed directly to Ray Serve. You can learn more about Ray Serve config files here: https://docs.ray.io/en/latest/serve/production-guide/config.html  # noqa: E501
+
+        :return: The ray_serve_config of this ProductionJobConfig.  # noqa: E501
+        :rtype: object
+        """
+        return self._ray_serve_config
+
+    @ray_serve_config.setter
+    def ray_serve_config(self, ray_serve_config):
+        """Sets the ray_serve_config of this ProductionJobConfig.
+
+        The Ray Serve config to use for this Production service. This config defines your Ray Serve application, and will be passed directly to Ray Serve. You can learn more about Ray Serve config files here: https://docs.ray.io/en/latest/serve/production-guide/config.html  # noqa: E501
+
+        :param ray_serve_config: The ray_serve_config of this ProductionJobConfig.  # noqa: E501
+        :type: object
+        """
+
+        self._ray_serve_config = ray_serve_config
+
+    @property
+    def runtime_env(self):
+        """Gets the runtime_env of this ProductionJobConfig.  # noqa: E501
+
+        A ray runtime env json. Your entrypoint will be run in the environment specified by this runtime env.  # noqa: E501
+
+        :return: The runtime_env of this ProductionJobConfig.  # noqa: E501
+        :rtype: RayRuntimeEnvConfig
+        """
+        return self._runtime_env
+
+    @runtime_env.setter
+    def runtime_env(self, runtime_env):
+        """Sets the runtime_env of this ProductionJobConfig.
+
+        A ray runtime env json. Your entrypoint will be run in the environment specified by this runtime env.  # noqa: E501
+
+        :param runtime_env: The runtime_env of this ProductionJobConfig.  # noqa: E501
+        :type: RayRuntimeEnvConfig
+        """
+
+        self._runtime_env = runtime_env
+
+    @property
+    def build_id(self):
+        """Gets the build_id of this ProductionJobConfig.  # noqa: E501
+
+        The id of the cluster env build. This id will determine the docker image your job is run on.  # noqa: E501
+
+        :return: The build_id of this ProductionJobConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._build_id
+
+    @build_id.setter
+    def build_id(self, build_id):
+        """Sets the build_id of this ProductionJobConfig.
+
+        The id of the cluster env build. This id will determine the docker image your job is run on.  # noqa: E501
+
+        :param build_id: The build_id of this ProductionJobConfig.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and build_id is None:  # noqa: E501
+            raise ValueError("Invalid value for `build_id`, must not be `None`")  # noqa: E501
+
+        self._build_id = build_id
+
+    @property
+    def compute_config_id(self):
+        """Gets the compute_config_id of this ProductionJobConfig.  # noqa: E501
+
+        The id of the compute configuration that you want to use. This id will specify the resources required for your job  # noqa: E501
+
+        :return: The compute_config_id of this ProductionJobConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._compute_config_id
+
+    @compute_config_id.setter
+    def compute_config_id(self, compute_config_id):
+        """Sets the compute_config_id of this ProductionJobConfig.
+
+        The id of the compute configuration that you want to use. This id will specify the resources required for your job  # noqa: E501
+
+        :param compute_config_id: The compute_config_id of this ProductionJobConfig.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and compute_config_id is None:  # noqa: E501
+            raise ValueError("Invalid value for `compute_config_id`, must not be `None`")  # noqa: E501
+
+        self._compute_config_id = compute_config_id
+
+    @property
+    def compute_config(self):
+        """Gets the compute_config of this ProductionJobConfig.  # noqa: E501
+
+        One-off compute that the cluster will use.  # noqa: E501
+
+        :return: The compute_config of this ProductionJobConfig.  # noqa: E501
+        :rtype: CreateClusterComputeConfig
+        """
+        return self._compute_config
+
+    @compute_config.setter
+    def compute_config(self, compute_config):
+        """Sets the compute_config of this ProductionJobConfig.
+
+        One-off compute that the cluster will use.  # noqa: E501
+
+        :param compute_config: The compute_config of this ProductionJobConfig.  # noqa: E501
+        :type: CreateClusterComputeConfig
+        """
+
+        self._compute_config = compute_config
+
+    @property
+    def max_retries(self):
+        """Gets the max_retries of this ProductionJobConfig.  # noqa: E501
+
+        The number of retries this job will attempt on failure. Set to None to set infinite retries  # noqa: E501
+
+        :return: The max_retries of this ProductionJobConfig.  # noqa: E501
+        :rtype: int
+        """
+        return self._max_retries
+
+    @max_retries.setter
+    def max_retries(self, max_retries):
+        """Sets the max_retries of this ProductionJobConfig.
+
+        The number of retries this job will attempt on failure. Set to None to set infinite retries  # noqa: E501
+
+        :param max_retries: The max_retries of this ProductionJobConfig.  # noqa: E501
+        :type: int
+        """
+
+        self._max_retries = max_retries
+
+    @property
+    def timeout_s(self):
+        """Gets the timeout_s of this ProductionJobConfig.  # noqa: E501
+
+        The timeout in seconds for each job run. Set to None for no limit to be set  # noqa: E501
+
+        :return: The timeout_s of this ProductionJobConfig.  # noqa: E501
+        :rtype: int
+        """
+        return self._timeout_s
+
+    @timeout_s.setter
+    def timeout_s(self, timeout_s):
+        """Sets the timeout_s of this ProductionJobConfig.
+
+        The timeout in seconds for each job run. Set to None for no limit to be set  # noqa: E501
+
+        :param timeout_s: The timeout_s of this ProductionJobConfig.  # noqa: E501
+        :type: int
+        """
+
+        self._timeout_s = timeout_s
+
+    @property
+    def runtime_env_config(self):
+        """Gets the runtime_env_config of this ProductionJobConfig.  # noqa: E501
+
+        DEPRECATED: Use runtime_env  # noqa: E501
+
+        :return: The runtime_env_config of this ProductionJobConfig.  # noqa: E501
+        :rtype: RayRuntimeEnvConfig
+        """
+        return self._runtime_env_config
+
+    @runtime_env_config.setter
+    def runtime_env_config(self, runtime_env_config):
+        """Sets the runtime_env_config of this ProductionJobConfig.
+
+        DEPRECATED: Use runtime_env  # noqa: E501
+
+        :param runtime_env_config: The runtime_env_config of this ProductionJobConfig.  # noqa: E501
+        :type: RayRuntimeEnvConfig
+        """
+
+        self._runtime_env_config = runtime_env_config
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = 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, ProductionJobConfig):
+            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, ProductionJobConfig):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/sdk/anyscale_client/models/production_job_state_transition.py

@@ -0,0 +1,293 @@
+# coding: utf-8
+
+"""
+    Anyscale API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from anyscale_client.configuration import Configuration
+
+
+class ProductionJobStateTransition(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    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.
+    """
+    openapi_types = {
+        'id': 'str',
+        'state_transitioned_at': 'datetime',
+        'current_state': 'HaJobStates',
+        'goal_state': 'HaJobGoalStates',
+        'error': 'str',
+        'operation_message': 'str',
+        'cluster_id': 'str'
+    }
+
+    attribute_map = {
+        'id': 'id',
+        'state_transitioned_at': 'state_transitioned_at',
+        'current_state': 'current_state',
+        'goal_state': 'goal_state',
+        'error': 'error',
+        'operation_message': 'operation_message',
+        'cluster_id': 'cluster_id'
+    }
+
+    def __init__(self, id=None, state_transitioned_at=None, current_state=None, goal_state=None, error=None, operation_message=None, cluster_id=None, local_vars_configuration=None):  # noqa: E501
+        """ProductionJobStateTransition - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+
+        self._id = None
+        self._state_transitioned_at = None
+        self._current_state = None
+        self._goal_state = None
+        self._error = None
+        self._operation_message = None
+        self._cluster_id = None
+        self.discriminator = None
+
+        self.id = id
+        self.state_transitioned_at = state_transitioned_at
+        self.current_state = current_state
+        if goal_state is not None:
+            self.goal_state = goal_state
+        if error is not None:
+            self.error = error
+        if operation_message is not None:
+            self.operation_message = operation_message
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
+
+    @property
+    def id(self):
+        """Gets the id of this ProductionJobStateTransition.  # noqa: E501
+
+        The id of this job state transition  # noqa: E501
+
+        :return: The id of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id):
+        """Sets the id of this ProductionJobStateTransition.
+
+        The id of this job state transition  # noqa: E501
+
+        :param id: The id of this ProductionJobStateTransition.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and id is None:  # noqa: E501
+            raise ValueError("Invalid value for `id`, must not be `None`")  # noqa: E501
+
+        self._id = id
+
+    @property
+    def state_transitioned_at(self):
+        """Gets the state_transitioned_at of this ProductionJobStateTransition.  # noqa: E501
+
+        The last time the state of this job was updated. This includes updates to the state and to the goal state  # noqa: E501
+
+        :return: The state_transitioned_at of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._state_transitioned_at
+
+    @state_transitioned_at.setter
+    def state_transitioned_at(self, state_transitioned_at):
+        """Sets the state_transitioned_at of this ProductionJobStateTransition.
+
+        The last time the state of this job was updated. This includes updates to the state and to the goal state  # noqa: E501
+
+        :param state_transitioned_at: The state_transitioned_at of this ProductionJobStateTransition.  # noqa: E501
+        :type: datetime
+        """
+        if self.local_vars_configuration.client_side_validation and state_transitioned_at is None:  # noqa: E501
+            raise ValueError("Invalid value for `state_transitioned_at`, must not be `None`")  # noqa: E501
+
+        self._state_transitioned_at = state_transitioned_at
+
+    @property
+    def current_state(self):
+        """Gets the current_state of this ProductionJobStateTransition.  # noqa: E501
+
+        The current state of the job  # noqa: E501
+
+        :return: The current_state of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: HaJobStates
+        """
+        return self._current_state
+
+    @current_state.setter
+    def current_state(self, current_state):
+        """Sets the current_state of this ProductionJobStateTransition.
+
+        The current state of the job  # noqa: E501
+
+        :param current_state: The current_state of this ProductionJobStateTransition.  # noqa: E501
+        :type: HaJobStates
+        """
+        if self.local_vars_configuration.client_side_validation and current_state is None:  # noqa: E501
+            raise ValueError("Invalid value for `current_state`, must not be `None`")  # noqa: E501
+
+        self._current_state = current_state
+
+    @property
+    def goal_state(self):
+        """Gets the goal_state of this ProductionJobStateTransition.  # noqa: E501
+
+        The goal state of the job  # noqa: E501
+
+        :return: The goal_state of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: HaJobGoalStates
+        """
+        return self._goal_state
+
+    @goal_state.setter
+    def goal_state(self, goal_state):
+        """Sets the goal_state of this ProductionJobStateTransition.
+
+        The goal state of the job  # noqa: E501
+
+        :param goal_state: The goal_state of this ProductionJobStateTransition.  # noqa: E501
+        :type: HaJobGoalStates
+        """
+
+        self._goal_state = goal_state
+
+    @property
+    def error(self):
+        """Gets the error of this ProductionJobStateTransition.  # noqa: E501
+
+        An error message that occurred in this job state transition  # noqa: E501
+
+        :return: The error of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: str
+        """
+        return self._error
+
+    @error.setter
+    def error(self, error):
+        """Sets the error of this ProductionJobStateTransition.
+
+        An error message that occurred in this job state transition  # noqa: E501
+
+        :param error: The error of this ProductionJobStateTransition.  # noqa: E501
+        :type: str
+        """
+
+        self._error = error
+
+    @property
+    def operation_message(self):
+        """Gets the operation_message of this ProductionJobStateTransition.  # noqa: E501
+
+        The logging message for this job state transition  # noqa: E501
+
+        :return: The operation_message of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: str
+        """
+        return self._operation_message
+
+    @operation_message.setter
+    def operation_message(self, operation_message):
+        """Sets the operation_message of this ProductionJobStateTransition.
+
+        The logging message for this job state transition  # noqa: E501
+
+        :param operation_message: The operation_message of this ProductionJobStateTransition.  # noqa: E501
+        :type: str
+        """
+
+        self._operation_message = operation_message
+
+    @property
+    def cluster_id(self):
+        """Gets the cluster_id of this ProductionJobStateTransition.  # noqa: E501
+
+        The id of the cluster the job is running on  # noqa: E501
+
+        :return: The cluster_id of this ProductionJobStateTransition.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id):
+        """Sets the cluster_id of this ProductionJobStateTransition.
+
+        The id of the cluster the job is running on  # noqa: E501
+
+        :param cluster_id: The cluster_id of this ProductionJobStateTransition.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = 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, ProductionJobStateTransition):
+            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, ProductionJobStateTransition):
+            return True
+
+        return self.to_dict() != other.to_dict()