anyscale 0.26.68__py3-none-any.whl → 0.26.70__py3-none-any.whl

anyscale/client/openapi_client/models/describe_machine_pool_machines_filters.py

@@ -34,15 +34,17 @@ class DescribeMachinePoolMachinesFilters(object):
     """
     openapi_types = {
         'allocation_states': 'list[MachineAllocationState]',
-        'partitions': 'list[str]'
+        'partitions': 'list[str]',
+        'machine_type_partition_filters': 'list[MachineTypePartitionFilter]'
     }
 
     attribute_map = {
         'allocation_states': 'allocation_states',
-        'partitions': 'partitions'
+        'partitions': 'partitions',
+        'machine_type_partition_filters': 'machine_type_partition_filters'
     }
 
-    def __init__(self, allocation_states=None, partitions=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, allocation_states=None, partitions=None, machine_type_partition_filters=None, local_vars_configuration=None):  # noqa: E501
         """DescribeMachinePoolMachinesFilters - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -50,12 +52,15 @@ class DescribeMachinePoolMachinesFilters(object):
 
         self._allocation_states = None
         self._partitions = None
+        self._machine_type_partition_filters = None
         self.discriminator = None
 
         if allocation_states is not None:
             self.allocation_states = allocation_states
         if partitions is not None:
             self.partitions = partitions
+        if machine_type_partition_filters is not None:
+            self.machine_type_partition_filters = machine_type_partition_filters
 
     @property
     def allocation_states(self):
@@ -84,7 +89,7 @@ class DescribeMachinePoolMachinesFilters(object):
     def partitions(self):
         """Gets the partitions of this DescribeMachinePoolMachinesFilters.  # noqa: E501
 
-        The partitions to filter by.  # noqa: E501
+        The partitions to filter by (for backward compatibility).  # noqa: E501
 
         :return: The partitions of this DescribeMachinePoolMachinesFilters.  # noqa: E501
         :rtype: list[str]
@@ -95,7 +100,7 @@ class DescribeMachinePoolMachinesFilters(object):
     def partitions(self, partitions):
         """Sets the partitions of this DescribeMachinePoolMachinesFilters.
 
-        The partitions to filter by.  # noqa: E501
+        The partitions to filter by (for backward compatibility).  # noqa: E501
 
         :param partitions: The partitions of this DescribeMachinePoolMachinesFilters.  # noqa: E501
         :type: list[str]
@@ -103,6 +108,29 @@ class DescribeMachinePoolMachinesFilters(object):
 
         self._partitions = partitions
 
+    @property
+    def machine_type_partition_filters(self):
+        """Gets the machine_type_partition_filters of this DescribeMachinePoolMachinesFilters.  # noqa: E501
+
+        The machine_type, partition pair to filter by.  # noqa: E501
+
+        :return: The machine_type_partition_filters of this DescribeMachinePoolMachinesFilters.  # noqa: E501
+        :rtype: list[MachineTypePartitionFilter]
+        """
+        return self._machine_type_partition_filters
+
+    @machine_type_partition_filters.setter
+    def machine_type_partition_filters(self, machine_type_partition_filters):
+        """Sets the machine_type_partition_filters of this DescribeMachinePoolMachinesFilters.
+
+        The machine_type, partition pair to filter by.  # noqa: E501
+
+        :param machine_type_partition_filters: The machine_type_partition_filters of this DescribeMachinePoolMachinesFilters.  # noqa: E501
+        :type: list[MachineTypePartitionFilter]
+        """
+
+        self._machine_type_partition_filters = machine_type_partition_filters
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

anyscale/client/openapi_client/models/describe_machine_pool_workloads_filters.py

@@ -34,15 +34,17 @@ class DescribeMachinePoolWorkloadsFilters(object):
     """
     openapi_types = {
         'workload_name': 'TextQuery',
-        'partitions': 'list[str]'
+        'partitions': 'list[str]',
+        'machine_type_partition_filters': 'list[MachineTypePartitionFilter]'
     }
 
     attribute_map = {
         'workload_name': 'workload_name',
-        'partitions': 'partitions'
+        'partitions': 'partitions',
+        'machine_type_partition_filters': 'machine_type_partition_filters'
     }
 
-    def __init__(self, workload_name=None, partitions=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, workload_name=None, partitions=None, machine_type_partition_filters=None, local_vars_configuration=None):  # noqa: E501
         """DescribeMachinePoolWorkloadsFilters - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -50,12 +52,15 @@ class DescribeMachinePoolWorkloadsFilters(object):
 
         self._workload_name = None
         self._partitions = None
+        self._machine_type_partition_filters = None
         self.discriminator = None
 
         if workload_name is not None:
             self.workload_name = workload_name
         if partitions is not None:
             self.partitions = partitions
+        if machine_type_partition_filters is not None:
+            self.machine_type_partition_filters = machine_type_partition_filters
 
     @property
     def workload_name(self):
@@ -84,7 +89,7 @@ class DescribeMachinePoolWorkloadsFilters(object):
     def partitions(self):
         """Gets the partitions of this DescribeMachinePoolWorkloadsFilters.  # noqa: E501
 
-        The partitions to filter workload machines by. If not provided, no filtering is performed.  # noqa: E501
+        The partitions to filter workload machines by (for backward compatibility).  # noqa: E501
 
         :return: The partitions of this DescribeMachinePoolWorkloadsFilters.  # noqa: E501
         :rtype: list[str]
@@ -95,7 +100,7 @@ class DescribeMachinePoolWorkloadsFilters(object):
     def partitions(self, partitions):
         """Sets the partitions of this DescribeMachinePoolWorkloadsFilters.
 
-        The partitions to filter workload machines by. If not provided, no filtering is performed.  # noqa: E501
+        The partitions to filter workload machines by (for backward compatibility).  # noqa: E501
 
         :param partitions: The partitions of this DescribeMachinePoolWorkloadsFilters.  # noqa: E501
         :type: list[str]
@@ -103,6 +108,29 @@ class DescribeMachinePoolWorkloadsFilters(object):
 
         self._partitions = partitions
 
+    @property
+    def machine_type_partition_filters(self):
+        """Gets the machine_type_partition_filters of this DescribeMachinePoolWorkloadsFilters.  # noqa: E501
+
+        The machine_type, partition pair to filter by.  # noqa: E501
+
+        :return: The machine_type_partition_filters of this DescribeMachinePoolWorkloadsFilters.  # noqa: E501
+        :rtype: list[MachineTypePartitionFilter]
+        """
+        return self._machine_type_partition_filters
+
+    @machine_type_partition_filters.setter
+    def machine_type_partition_filters(self, machine_type_partition_filters):
+        """Sets the machine_type_partition_filters of this DescribeMachinePoolWorkloadsFilters.
+
+        The machine_type, partition pair to filter by.  # noqa: E501
+
+        :param machine_type_partition_filters: The machine_type_partition_filters of this DescribeMachinePoolWorkloadsFilters.  # noqa: E501
+        :type: list[MachineTypePartitionFilter]
+        """
+
+        self._machine_type_partition_filters = machine_type_partition_filters
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

anyscale/client/openapi_client/models/{service_event_level.py → entity_type.py}

@@ -18,7 +18,7 @@ import six
 from openapi_client.configuration import Configuration
 
 
-class ServiceEventLevel(object):
+class EntityType(object):
     """NOTE: This class is auto generated by OpenAPI Generator.
     Ref: https://openapi-generator.tech
 
@@ -28,12 +28,12 @@ class ServiceEventLevel(object):
     """
     allowed enum values
     """
-    ERROR = "ERROR"
-    INFO = "INFO"
-    WARNING = "WARNING"
-    TRACE = "TRACE"
+    PRODUCTION_JOB = "production_job"
+    WORKSPACE = "workspace"
+    SERVICE = "service"
+    SESSION = "session"
 
-    allowable_values = [ERROR, INFO, WARNING, TRACE]  # noqa: E501
+    allowable_values = [PRODUCTION_JOB, WORKSPACE, SERVICE, SESSION]  # noqa: E501
 
     """
     Attributes:
@@ -49,7 +49,7 @@ class ServiceEventLevel(object):
     }
 
     def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """ServiceEventLevel - a model defined in OpenAPI"""  # noqa: E501
+        """EntityType - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
         self.local_vars_configuration = local_vars_configuration
@@ -89,14 +89,14 @@ class ServiceEventLevel(object):
 
     def __eq__(self, other):
         """Returns true if both objects are equal"""
-        if not isinstance(other, ServiceEventLevel):
+        if not isinstance(other, EntityType):
             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, ServiceEventLevel):
+        if not isinstance(other, EntityType):
             return True
 
         return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/event_level.py

@@ -33,9 +33,10 @@ class EventLevel(object):
     ERROR = "ERROR"
     TRACE = "TRACE"
     DEBUG = "DEBUG"
+    CRITICAL = "CRITICAL"
     UNKNOWN = "UNKNOWN"
 
-    allowable_values = [INFO, WARNING, ERROR, TRACE, DEBUG, UNKNOWN]  # noqa: E501
+    allowable_values = [INFO, WARNING, ERROR, TRACE, DEBUG, CRITICAL, UNKNOWN]  # noqa: E501
 
     """
     Attributes:

anyscale/client/openapi_client/models/job_event_fields.py

@@ -0,0 +1,206 @@
+# 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 JobEventFields(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 = {
+        'ha_job_id': 'str',
+        'error_type': 'HaJobErrorTypes',
+        'has_additional_sensitive_message': 'bool',
+        'event_type': 'str'
+    }
+
+    attribute_map = {
+        'ha_job_id': 'ha_job_id',
+        'error_type': 'error_type',
+        'has_additional_sensitive_message': 'has_additional_sensitive_message',
+        'event_type': 'event_type'
+    }
+
+    def __init__(self, ha_job_id=None, error_type=None, has_additional_sensitive_message=False, event_type=None, local_vars_configuration=None):  # noqa: E501
+        """JobEventFields - 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._ha_job_id = None
+        self._error_type = None
+        self._has_additional_sensitive_message = None
+        self._event_type = None
+        self.discriminator = None
+
+        if ha_job_id is not None:
+            self.ha_job_id = ha_job_id
+        if error_type is not None:
+            self.error_type = error_type
+        if has_additional_sensitive_message is not None:
+            self.has_additional_sensitive_message = has_additional_sensitive_message
+        if event_type is not None:
+            self.event_type = event_type
+
+    @property
+    def ha_job_id(self):
+        """Gets the ha_job_id of this JobEventFields.  # noqa: E501
+
+        The id of the Job.  # noqa: E501
+
+        :return: The ha_job_id of this JobEventFields.  # noqa: E501
+        :rtype: str
+        """
+        return self._ha_job_id
+
+    @ha_job_id.setter
+    def ha_job_id(self, ha_job_id):
+        """Sets the ha_job_id of this JobEventFields.
+
+        The id of the Job.  # noqa: E501
+
+        :param ha_job_id: The ha_job_id of this JobEventFields.  # noqa: E501
+        :type: str
+        """
+
+        self._ha_job_id = ha_job_id
+
+    @property
+    def error_type(self):
+        """Gets the error_type of this JobEventFields.  # noqa: E501
+
+        Type of Job error. Only populated for JOB type events that are in error state  # noqa: E501
+
+        :return: The error_type of this JobEventFields.  # noqa: E501
+        :rtype: HaJobErrorTypes
+        """
+        return self._error_type
+
+    @error_type.setter
+    def error_type(self, error_type):
+        """Sets the error_type of this JobEventFields.
+
+        Type of Job error. Only populated for JOB type events that are in error state  # noqa: E501
+
+        :param error_type: The error_type of this JobEventFields.  # noqa: E501
+        :type: HaJobErrorTypes
+        """
+
+        self._error_type = error_type
+
+    @property
+    def has_additional_sensitive_message(self):
+        """Gets the has_additional_sensitive_message of this JobEventFields.  # noqa: E501
+
+        Whether we have additional sensitive message details stored that can be subsequently fetched.  # noqa: E501
+
+        :return: The has_additional_sensitive_message of this JobEventFields.  # noqa: E501
+        :rtype: bool
+        """
+        return self._has_additional_sensitive_message
+
+    @has_additional_sensitive_message.setter
+    def has_additional_sensitive_message(self, has_additional_sensitive_message):
+        """Sets the has_additional_sensitive_message of this JobEventFields.
+
+        Whether we have additional sensitive message details stored that can be subsequently fetched.  # noqa: E501
+
+        :param has_additional_sensitive_message: The has_additional_sensitive_message of this JobEventFields.  # noqa: E501
+        :type: bool
+        """
+
+        self._has_additional_sensitive_message = has_additional_sensitive_message
+
+    @property
+    def event_type(self):
+        """Gets the event_type of this JobEventFields.  # noqa: E501
+
+        Type of job event  # noqa: E501
+
+        :return: The event_type of this JobEventFields.  # noqa: E501
+        :rtype: str
+        """
+        return self._event_type
+
+    @event_type.setter
+    def event_type(self, event_type):
+        """Sets the event_type of this JobEventFields.
+
+        Type of job event  # noqa: E501
+
+        :param event_type: The event_type of this JobEventFields.  # noqa: E501
+        :type: str
+        """
+
+        self._event_type = event_type
+
+    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, JobEventFields):
+            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, JobEventFields):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/lineage_graph_node.py

@@ -41,10 +41,11 @@ class LineageGraphNode(object):
         'updated_at': 'datetime',
         'created_by': 'str',
         'updated_by': 'str',
-        'node_type': 'LineageNodeType',
+        'workload_type': 'LineageWorkloadType',
+        'anyscale_workload_id': 'str',
         'artifact_type': 'LineageArtifactType',
         'artifact_file_format': 'str',
-        'workload_type': 'LineageWorkloadType'
+        'node_type': 'LineageNodeType'
     }
 
     attribute_map = {
@@ -56,13 +57,14 @@ class LineageGraphNode(object):
         'updated_at': 'updated_at',
         'created_by': 'created_by',
         'updated_by': 'updated_by',
-        'node_type': 'node_type',
+        'workload_type': 'workload_type',
+        'anyscale_workload_id': 'anyscale_workload_id',
         'artifact_type': 'artifact_type',
         'artifact_file_format': 'artifact_file_format',
-        'workload_type': 'workload_type'
+        'node_type': 'node_type'
     }
 
-    def __init__(self, id=None, name=None, namespace_name=None, uri=None, created_at=None, updated_at=None, created_by=None, updated_by=None, node_type=None, artifact_type=None, artifact_file_format=None, workload_type=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, id=None, name=None, namespace_name=None, uri=None, created_at=None, updated_at=None, created_by=None, updated_by=None, workload_type=None, anyscale_workload_id=None, artifact_type=None, artifact_file_format=None, node_type=None, local_vars_configuration=None):  # noqa: E501
         """LineageGraphNode - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -76,10 +78,11 @@ class LineageGraphNode(object):
         self._updated_at = None
         self._created_by = None
         self._updated_by = None
-        self._node_type = None
+        self._workload_type = None
+        self._anyscale_workload_id = None
         self._artifact_type = None
         self._artifact_file_format = None
-        self._workload_type = None
+        self._node_type = None
         self.discriminator = None
 
         self.id = id
@@ -90,13 +93,15 @@ class LineageGraphNode(object):
         self.updated_at = updated_at
         self.created_by = created_by
         self.updated_by = updated_by
-        self.node_type = node_type
+        if workload_type is not None:
+            self.workload_type = workload_type
+        if anyscale_workload_id is not None:
+            self.anyscale_workload_id = anyscale_workload_id
         if artifact_type is not None:
             self.artifact_type = artifact_type
         if artifact_file_format is not None:
             self.artifact_file_format = artifact_file_format
-        if workload_type is not None:
-            self.workload_type = workload_type
+        self.node_type = node_type
 
     @property
     def id(self):
@@ -299,35 +304,56 @@ class LineageGraphNode(object):
         self._updated_by = updated_by
 
     @property
-    def node_type(self):
-        """Gets the node_type of this LineageGraphNode.  # noqa: E501
+    def workload_type(self):
+        """Gets the workload_type of this LineageGraphNode.  # noqa: E501
 
-        Type of node (ARTIFACT/WORKLOAD)  # noqa: E501
+        Type of workload (WORKSPACE/SERVICE/JOB)  # noqa: E501
 
-        :return: The node_type of this LineageGraphNode.  # noqa: E501
-        :rtype: LineageNodeType
+        :return: The workload_type of this LineageGraphNode.  # noqa: E501
+        :rtype: LineageWorkloadType
         """
-        return self._node_type
+        return self._workload_type
 
-    @node_type.setter
-    def node_type(self, node_type):
-        """Sets the node_type of this LineageGraphNode.
+    @workload_type.setter
+    def workload_type(self, workload_type):
+        """Sets the workload_type of this LineageGraphNode.
 
-        Type of node (ARTIFACT/WORKLOAD)  # noqa: E501
+        Type of workload (WORKSPACE/SERVICE/JOB)  # noqa: E501
 
-        :param node_type: The node_type of this LineageGraphNode.  # noqa: E501
-        :type: LineageNodeType
+        :param workload_type: The workload_type of this LineageGraphNode.  # noqa: E501
+        :type: LineageWorkloadType
         """
-        if self.local_vars_configuration.client_side_validation and node_type is None:  # noqa: E501
-            raise ValueError("Invalid value for `node_type`, must not be `None`")  # noqa: E501
 
-        self._node_type = node_type
+        self._workload_type = workload_type
+
+    @property
+    def anyscale_workload_id(self):
+        """Gets the anyscale_workload_id of this LineageGraphNode.  # noqa: E501
+
+        Anyscale workload ID  # noqa: E501
+
+        :return: The anyscale_workload_id of this LineageGraphNode.  # noqa: E501
+        :rtype: str
+        """
+        return self._anyscale_workload_id
+
+    @anyscale_workload_id.setter
+    def anyscale_workload_id(self, anyscale_workload_id):
+        """Sets the anyscale_workload_id of this LineageGraphNode.
+
+        Anyscale workload ID  # noqa: E501
+
+        :param anyscale_workload_id: The anyscale_workload_id of this LineageGraphNode.  # noqa: E501
+        :type: str
+        """
+
+        self._anyscale_workload_id = anyscale_workload_id
 
     @property
     def artifact_type(self):
         """Gets the artifact_type of this LineageGraphNode.  # noqa: E501
 
-        Type of artifact if node_type is ARTIFACT  # noqa: E501
+        Type of artifact (DATASET/MODEL)  # noqa: E501
 
         :return: The artifact_type of this LineageGraphNode.  # noqa: E501
         :rtype: LineageArtifactType
@@ -338,7 +364,7 @@ class LineageGraphNode(object):
     def artifact_type(self, artifact_type):
         """Sets the artifact_type of this LineageGraphNode.
 
-        Type of artifact if node_type is ARTIFACT  # noqa: E501
+        Type of artifact (DATASET/MODEL)  # noqa: E501
 
         :param artifact_type: The artifact_type of this LineageGraphNode.  # noqa: E501
         :type: LineageArtifactType
@@ -350,7 +376,7 @@ class LineageGraphNode(object):
     def artifact_file_format(self):
         """Gets the artifact_file_format of this LineageGraphNode.  # noqa: E501
 
-        File format if artifact  # noqa: E501
+        File format (e.g., parquet, csv)  # noqa: E501
 
         :return: The artifact_file_format of this LineageGraphNode.  # noqa: E501
         :rtype: str
@@ -361,7 +387,7 @@ class LineageGraphNode(object):
     def artifact_file_format(self, artifact_file_format):
         """Sets the artifact_file_format of this LineageGraphNode.
 
-        File format if artifact  # noqa: E501
+        File format (e.g., parquet, csv)  # noqa: E501
 
         :param artifact_file_format: The artifact_file_format of this LineageGraphNode.  # noqa: E501
         :type: str
@@ -370,27 +396,29 @@ class LineageGraphNode(object):
         self._artifact_file_format = artifact_file_format
 
     @property
-    def workload_type(self):
-        """Gets the workload_type of this LineageGraphNode.  # noqa: E501
+    def node_type(self):
+        """Gets the node_type of this LineageGraphNode.  # noqa: E501
 
-        Type of workload if node_type is WORKLOAD  # noqa: E501
+        Type of node (ARTIFACT/WORKLOAD)  # noqa: E501
 
-        :return: The workload_type of this LineageGraphNode.  # noqa: E501
-        :rtype: LineageWorkloadType
+        :return: The node_type of this LineageGraphNode.  # noqa: E501
+        :rtype: LineageNodeType
         """
-        return self._workload_type
+        return self._node_type
 
-    @workload_type.setter
-    def workload_type(self, workload_type):
-        """Sets the workload_type of this LineageGraphNode.
+    @node_type.setter
+    def node_type(self, node_type):
+        """Sets the node_type of this LineageGraphNode.
 
-        Type of workload if node_type is WORKLOAD  # noqa: E501
+        Type of node (ARTIFACT/WORKLOAD)  # noqa: E501
 
-        :param workload_type: The workload_type of this LineageGraphNode.  # noqa: E501
-        :type: LineageWorkloadType
+        :param node_type: The node_type of this LineageGraphNode.  # noqa: E501
+        :type: LineageNodeType
         """
+        if self.local_vars_configuration.client_side_validation and node_type is None:  # noqa: E501
+            raise ValueError("Invalid value for `node_type`, must not be `None`")  # noqa: E501
 
-        self._workload_type = workload_type
+        self._node_type = node_type
 
     def to_dict(self):
         """Returns the model properties as a dict"""