anyscale 0.24.88__py3-none-any.whl → 0.25.0__py3-none-any.whl

anyscale/client/openapi_client/models/request_state_info.py

@@ -0,0 +1,210 @@
+# coding: utf-8
+
+"""
+    Managed Ray 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 openapi_client.configuration import Configuration
+
+
+class RequestStateInfo(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 = {
+        'size': 'int',
+        'machine_type': 'str',
+        'workload_info': 'WorkloadInfo',
+        'partition_scores': 'dict(str, int)'
+    }
+
+    attribute_map = {
+        'size': 'size',
+        'machine_type': 'machine_type',
+        'workload_info': 'workload_info',
+        'partition_scores': 'partition_scores'
+    }
+
+    def __init__(self, size=None, machine_type=None, workload_info=None, partition_scores=None, local_vars_configuration=None):  # noqa: E501
+        """RequestStateInfo - 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._size = None
+        self._machine_type = None
+        self._workload_info = None
+        self._partition_scores = None
+        self.discriminator = None
+
+        self.size = size
+        self.machine_type = machine_type
+        self.workload_info = workload_info
+        self.partition_scores = partition_scores
+
+    @property
+    def size(self):
+        """Gets the size of this RequestStateInfo.  # noqa: E501
+
+        The size of the request.  # noqa: E501
+
+        :return: The size of this RequestStateInfo.  # noqa: E501
+        :rtype: int
+        """
+        return self._size
+
+    @size.setter
+    def size(self, size):
+        """Sets the size of this RequestStateInfo.
+
+        The size of the request.  # noqa: E501
+
+        :param size: The size of this RequestStateInfo.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and size is None:  # noqa: E501
+            raise ValueError("Invalid value for `size`, must not be `None`")  # noqa: E501
+
+        self._size = size
+
+    @property
+    def machine_type(self):
+        """Gets the machine_type of this RequestStateInfo.  # noqa: E501
+
+        The type of the machine.  # noqa: E501
+
+        :return: The machine_type of this RequestStateInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._machine_type
+
+    @machine_type.setter
+    def machine_type(self, machine_type):
+        """Sets the machine_type of this RequestStateInfo.
+
+        The type of the machine.  # noqa: E501
+
+        :param machine_type: The machine_type of this RequestStateInfo.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and machine_type is None:  # noqa: E501
+            raise ValueError("Invalid value for `machine_type`, must not be `None`")  # noqa: E501
+
+        self._machine_type = machine_type
+
+    @property
+    def workload_info(self):
+        """Gets the workload_info of this RequestStateInfo.  # noqa: E501
+
+        Workload metadata associated with this request.  # noqa: E501
+
+        :return: The workload_info of this RequestStateInfo.  # noqa: E501
+        :rtype: WorkloadInfo
+        """
+        return self._workload_info
+
+    @workload_info.setter
+    def workload_info(self, workload_info):
+        """Sets the workload_info of this RequestStateInfo.
+
+        Workload metadata associated with this request.  # noqa: E501
+
+        :param workload_info: The workload_info of this RequestStateInfo.  # noqa: E501
+        :type: WorkloadInfo
+        """
+        if self.local_vars_configuration.client_side_validation and workload_info is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_info`, must not be `None`")  # noqa: E501
+
+        self._workload_info = workload_info
+
+    @property
+    def partition_scores(self):
+        """Gets the partition_scores of this RequestStateInfo.  # noqa: E501
+
+        The scores of the partitions.  # noqa: E501
+
+        :return: The partition_scores of this RequestStateInfo.  # noqa: E501
+        :rtype: dict(str, int)
+        """
+        return self._partition_scores
+
+    @partition_scores.setter
+    def partition_scores(self, partition_scores):
+        """Sets the partition_scores of this RequestStateInfo.
+
+        The scores of the partitions.  # noqa: E501
+
+        :param partition_scores: The partition_scores of this RequestStateInfo.  # noqa: E501
+        :type: dict(str, int)
+        """
+        if self.local_vars_configuration.client_side_validation and partition_scores is None:  # noqa: E501
+            raise ValueError("Invalid value for `partition_scores`, must not be `None`")  # noqa: E501
+
+        self._partition_scores = partition_scores
+
+    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, RequestStateInfo):
+            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, RequestStateInfo):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/{sessionevent_list_response.py → scheduler_info.py}

@@ -18,7 +18,7 @@ import six
 from openapi_client.configuration import Configuration
 
 
-class SessioneventListResponse(object):
+class SchedulerInfo(object):
     """NOTE: This class is auto generated by OpenAPI Generator.
     Ref: https://openapi-generator.tech
 
@@ -33,72 +33,77 @@ class SessioneventListResponse(object):
                             and the value is json key in definition.
     """
     openapi_types = {
-        'results': 'list[SessionEvent]',
-        'metadata': 'ListResponseMetadata'
+        'requests': 'list[RequestStateInfo]',
+        'machines': 'list[MachineStateInfo]'
     }
 
     attribute_map = {
-        'results': 'results',
-        'metadata': 'metadata'
+        'requests': 'requests',
+        'machines': 'machines'
     }
 
-    def __init__(self, results=None, metadata=None, local_vars_configuration=None):  # noqa: E501
-        """SessioneventListResponse - a model defined in OpenAPI"""  # noqa: E501
+    def __init__(self, requests=None, machines=None, local_vars_configuration=None):  # noqa: E501
+        """SchedulerInfo - 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._results = None
-        self._metadata = None
+        self._requests = None
+        self._machines = None
         self.discriminator = None
 
-        self.results = results
-        if metadata is not None:
-            self.metadata = metadata
+        self.requests = requests
+        self.machines = machines
 
     @property
-    def results(self):
-        """Gets the results of this SessioneventListResponse.  # noqa: E501
+    def requests(self):
+        """Gets the requests of this SchedulerInfo.  # noqa: E501
 
+        Current request states.  # noqa: E501
 
-        :return: The results of this SessioneventListResponse.  # noqa: E501
-        :rtype: list[SessionEvent]
+        :return: The requests of this SchedulerInfo.  # noqa: E501
+        :rtype: list[RequestStateInfo]
         """
-        return self._results
+        return self._requests
 
-    @results.setter
-    def results(self, results):
-        """Sets the results of this SessioneventListResponse.
+    @requests.setter
+    def requests(self, requests):
+        """Sets the requests of this SchedulerInfo.
 
+        Current request states.  # noqa: E501
 
-        :param results: The results of this SessioneventListResponse.  # noqa: E501
-        :type: list[SessionEvent]
+        :param requests: The requests of this SchedulerInfo.  # noqa: E501
+        :type: list[RequestStateInfo]
         """
-        if self.local_vars_configuration.client_side_validation and results is None:  # noqa: E501
-            raise ValueError("Invalid value for `results`, must not be `None`")  # noqa: E501
+        if self.local_vars_configuration.client_side_validation and requests is None:  # noqa: E501
+            raise ValueError("Invalid value for `requests`, must not be `None`")  # noqa: E501
 
-        self._results = results
+        self._requests = requests
 
     @property
-    def metadata(self):
-        """Gets the metadata of this SessioneventListResponse.  # noqa: E501
+    def machines(self):
+        """Gets the machines of this SchedulerInfo.  # noqa: E501
 
+        Current machine states.  # noqa: E501
 
-        :return: The metadata of this SessioneventListResponse.  # noqa: E501
-        :rtype: ListResponseMetadata
+        :return: The machines of this SchedulerInfo.  # noqa: E501
+        :rtype: list[MachineStateInfo]
         """
-        return self._metadata
+        return self._machines
 
-    @metadata.setter
-    def metadata(self, metadata):
-        """Sets the metadata of this SessioneventListResponse.
+    @machines.setter
+    def machines(self, machines):
+        """Sets the machines of this SchedulerInfo.
 
+        Current machine states.  # noqa: E501
 
-        :param metadata: The metadata of this SessioneventListResponse.  # noqa: E501
-        :type: ListResponseMetadata
+        :param machines: The machines of this SchedulerInfo.  # noqa: E501
+        :type: list[MachineStateInfo]
         """
+        if self.local_vars_configuration.client_side_validation and machines is None:  # noqa: E501
+            raise ValueError("Invalid value for `machines`, must not be `None`")  # noqa: E501
 
-        self._metadata = metadata
+        self._machines = machines
 
     def to_dict(self):
         """Returns the model properties as a dict"""
@@ -134,14 +139,14 @@ class SessioneventListResponse(object):
 
     def __eq__(self, other):
         """Returns true if both objects are equal"""
-        if not isinstance(other, SessioneventListResponse):
+        if not isinstance(other, SchedulerInfo):
             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, SessioneventListResponse):
+        if not isinstance(other, SchedulerInfo):
             return True
 
         return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/usage_by_cluster.py

@@ -37,6 +37,7 @@ class UsageByCluster(object):
         'date': 'date',
         'user_id': 'str',
         'user_email': 'str',
+        'user_name': 'str',
         'cloud_id': 'str',
         'cloud_name': 'str',
         'project_id': 'str',
@@ -57,6 +58,7 @@ class UsageByCluster(object):
         'date': 'date',
         'user_id': 'user_id',
         'user_email': 'user_email',
+        'user_name': 'user_name',
         'cloud_id': 'cloud_id',
         'cloud_name': 'cloud_name',
         'project_id': 'project_id',
@@ -72,7 +74,7 @@ class UsageByCluster(object):
         'workspace_name': 'workspace_name'
     }
 
-    def __init__(self, anyscale_credits=None, date=None, user_id=None, user_email=None, cloud_id=None, cloud_name=None, project_id=None, project_name=None, cluster_id=None, job_id=None, job_name=None, job_queue_id=None, job_queue_name=None, service_id=None, service_name=None, workspace_id=None, workspace_name=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, anyscale_credits=None, date=None, user_id=None, user_email=None, user_name=None, cloud_id=None, cloud_name=None, project_id=None, project_name=None, cluster_id=None, job_id=None, job_name=None, job_queue_id=None, job_queue_name=None, service_id=None, service_name=None, workspace_id=None, workspace_name=None, local_vars_configuration=None):  # noqa: E501
         """UsageByCluster - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -82,6 +84,7 @@ class UsageByCluster(object):
         self._date = None
         self._user_id = None
         self._user_email = None
+        self._user_name = None
         self._cloud_id = None
         self._cloud_name = None
         self._project_id = None
@@ -102,6 +105,7 @@ class UsageByCluster(object):
             self.date = date
         self.user_id = user_id
         self.user_email = user_email
+        self.user_name = user_name
         self.cloud_id = cloud_id
         self.cloud_name = cloud_name
         self.project_id = project_id
@@ -214,6 +218,29 @@ class UsageByCluster(object):
 
         self._user_email = user_email
 
+    @property
+    def user_name(self):
+        """Gets the user_name of this UsageByCluster.  # noqa: E501
+
+
+        :return: The user_name of this UsageByCluster.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_name
+
+    @user_name.setter
+    def user_name(self, user_name):
+        """Sets the user_name of this UsageByCluster.
+
+
+        :param user_name: The user_name of this UsageByCluster.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and user_name is None:  # noqa: E501
+            raise ValueError("Invalid value for `user_name`, must not be `None`")  # noqa: E501
+
+        self._user_name = user_name
+
     @property
     def cloud_id(self):
         """Gets the cloud_id of this UsageByCluster.  # noqa: E501

anyscale/client/openapi_client/models/usage_by_user.py

@@ -36,17 +36,19 @@ class UsageByUser(object):
         'anyscale_credits': 'float',
         'date': 'date',
         'user_id': 'str',
-        'user_email': 'str'
+        'user_email': 'str',
+        'user_name': 'str'
     }
 
     attribute_map = {
         'anyscale_credits': 'anyscale_credits',
         'date': 'date',
         'user_id': 'user_id',
-        'user_email': 'user_email'
+        'user_email': 'user_email',
+        'user_name': 'user_name'
     }
 
-    def __init__(self, anyscale_credits=None, date=None, user_id=None, user_email=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, anyscale_credits=None, date=None, user_id=None, user_email=None, user_name=None, local_vars_configuration=None):  # noqa: E501
         """UsageByUser - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -56,6 +58,7 @@ class UsageByUser(object):
         self._date = None
         self._user_id = None
         self._user_email = None
+        self._user_name = None
         self.discriminator = None
 
         self.anyscale_credits = anyscale_credits
@@ -63,6 +66,7 @@ class UsageByUser(object):
             self.date = date
         self.user_id = user_id
         self.user_email = user_email
+        self.user_name = user_name
 
     @property
     def anyscale_credits(self):
@@ -154,6 +158,29 @@ class UsageByUser(object):
 
         self._user_email = user_email
 
+    @property
+    def user_name(self):
+        """Gets the user_name of this UsageByUser.  # noqa: E501
+
+
+        :return: The user_name of this UsageByUser.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_name
+
+    @user_name.setter
+    def user_name(self, user_name):
+        """Sets the user_name of this UsageByUser.
+
+
+        :param user_name: The user_name of this UsageByUser.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and user_name is None:  # noqa: E501
+            raise ValueError("Invalid value for `user_name`, must not be `None`")  # noqa: E501
+
+        self._user_name = user_name
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}