anyscale 0.26.0__py3-none-any.whl → 0.26.2__py3-none-any.whl

anyscale/client/openapi_client/models/__init__.py

@@ -17,6 +17,7 @@ from __future__ import absolute_import
 from openapi_client.models.aws_credentials import AWSCredentials
 from openapi_client.models.aws_memory_db_cluster_config import AWSMemoryDBClusterConfig
 from openapi_client.models.access_config import AccessConfig
+from openapi_client.models.actor_status import ActorStatus
 from openapi_client.models.admin_create_user import AdminCreateUser
 from openapi_client.models.admin_created_user import AdminCreatedUser
 from openapi_client.models.admincreateduser_list_response import AdmincreateduserListResponse
@@ -202,7 +203,7 @@ from openapi_client.models.create_otp_return_api_model import CreateOTPReturnApi
 from openapi_client.models.create_organization_configuration import CreateOrganizationConfiguration
 from openapi_client.models.create_organization_invitation import CreateOrganizationInvitation
 from openapi_client.models.create_production_job_config import CreateProductionJobConfig
-from openapi_client.models.create_resource_alert import CreateResourceAlert
+from openapi_client.models.create_resource_notification import CreateResourceNotification
 from openapi_client.models.create_resource_quota import CreateResourceQuota
 from openapi_client.models.create_schedule import CreateSchedule
 from openapi_client.models.create_session_from_snapshot_options import CreateSessionFromSnapshotOptions
@@ -220,7 +221,6 @@ from openapi_client.models.createotpreturnapimodel_response import Createotpretu
 from openapi_client.models.createsessionresponse_response import CreatesessionresponseResponse
 from openapi_client.models.credit_card_information import CreditCardInformation
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
-from openapi_client.models.customer_billing_type import CustomerBillingType
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.dataset import Dataset
 from openapi_client.models.dataset_list_response import DatasetListResponse
@@ -346,6 +346,7 @@ from openapi_client.models.invoice_list_response import InvoiceListResponse
 from openapi_client.models.invoice_status import InvoiceStatus
 from openapi_client.models.invoices_query import InvoicesQuery
 from openapi_client.models.job_access import JobAccess
+from openapi_client.models.job_details import JobDetails
 from openapi_client.models.job_queue import JobQueue
 from openapi_client.models.job_queue_config import JobQueueConfig
 from openapi_client.models.job_queue_execution_mode import JobQueueExecutionMode
@@ -370,7 +371,7 @@ from openapi_client.models.lb_resource import LBResource
 from openapi_client.models.lbresource_response import LbresourceResponse
 from openapi_client.models.list_machine_pools_response import ListMachinePoolsResponse
 from openapi_client.models.list_machines_response import ListMachinesResponse
-from openapi_client.models.list_resource_alerts_query import ListResourceAlertsQuery
+from openapi_client.models.list_resource_notifications_query import ListResourceNotificationsQuery
 from openapi_client.models.list_resource_quotas_query import ListResourceQuotasQuery
 from openapi_client.models.list_response_metadata import ListResponseMetadata
 from openapi_client.models.listmachinepoolsresponse_response import ListmachinepoolsresponseResponse
@@ -397,6 +398,7 @@ from openapi_client.models.longrunningworkload_list_response import Longrunningw
 from openapi_client.models.machine_allocation_state import MachineAllocationState
 from openapi_client.models.machine_connection_state import MachineConnectionState
 from openapi_client.models.machine_info import MachineInfo
+from openapi_client.models.machine_launch_failure import MachineLaunchFailure
 from openapi_client.models.machine_pool import MachinePool
 from openapi_client.models.machine_state_info import MachineStateInfo
 from openapi_client.models.metronome_customer_info_model import MetronomeCustomerInfoModel
@@ -506,17 +508,19 @@ from openapi_client.models.request_state_info import RequestStateInfo
 from openapi_client.models.requestemailmagiclinkresponse_response import RequestemailmagiclinkresponseResponse
 from openapi_client.models.requestotpreturnapimodel_response import RequestotpreturnapimodelResponse
 from openapi_client.models.reset_password_params import ResetPasswordParams
-from openapi_client.models.resource_alert import ResourceAlert
 from openapi_client.models.resource_alert_event_type import ResourceAlertEventType
+from openapi_client.models.resource_notification import ResourceNotification
 from openapi_client.models.resource_quota import ResourceQuota
 from openapi_client.models.resource_quota_status import ResourceQuotaStatus
-from openapi_client.models.resourcealert_list_response import ResourcealertListResponse
-from openapi_client.models.resourcealert_response import ResourcealertResponse
+from openapi_client.models.resourcenotification_list_response import ResourcenotificationListResponse
+from openapi_client.models.resourcenotification_response import ResourcenotificationResponse
 from openapi_client.models.resourcequota_list_response import ResourcequotaListResponse
 from openapi_client.models.resourcequota_response import ResourcequotaResponse
 from openapi_client.models.resources import Resources
 from openapi_client.models.rollback_service_model import RollbackServiceModel
 from openapi_client.models.rollout_strategy import RolloutStrategy
+from openapi_client.models.run_attempt_status import RunAttemptStatus
+from openapi_client.models.run_status import RunStatus
 from openapi_client.models.s3_download_location import S3DownloadLocation
 from openapi_client.models.sso_login_info import SSOLoginInfo
 from openapi_client.models.supportedbaseimagesenum import SUPPORTEDBASEIMAGESENUM
@@ -602,6 +606,11 @@ from openapi_client.models.timestamped_logs_output import TimestampedLogsOutput
 from openapi_client.models.timestampedlogsoutput_response import TimestampedlogsoutputResponse
 from openapi_client.models.tool import Tool
 from openapi_client.models.tracing_config import TracingConfig
+from openapi_client.models.train_resources import TrainResources
+from openapi_client.models.train_run import TrainRun
+from openapi_client.models.train_run_attempt import TrainRunAttempt
+from openapi_client.models.train_worker import TrainWorker
+from openapi_client.models.trainrun_list_response import TrainrunListResponse
 from openapi_client.models.try_login_email_response import TryLoginEmailResponse
 from openapi_client.models.tryloginemailresponse_response import TryloginemailresponseResponse
 from openapi_client.models.ux_instance import UXInstance

anyscale/client/openapi_client/models/{customer_billing_type.py → actor_status.py}

@@ -18,7 +18,7 @@ import six
 from openapi_client.configuration import Configuration
 
 
-class CustomerBillingType(object):
+class ActorStatus(object):
     """NOTE: This class is auto generated by OpenAPI Generator.
     Ref: https://openapi-generator.tech
 
@@ -28,11 +28,10 @@ class CustomerBillingType(object):
     """
     allowed enum values
     """
-    SELF_SERVICE = "SELF_SERVICE"
-    PROOF_OF_CONCEPT = "PROOF_OF_CONCEPT"
-    EXTERNAL_TRAINING = "EXTERNAL_TRAINING"
+    ALIVE = "ALIVE"
+    DEAD = "DEAD"
 
-    allowable_values = [SELF_SERVICE, PROOF_OF_CONCEPT, EXTERNAL_TRAINING]  # noqa: E501
+    allowable_values = [ALIVE, DEAD]  # noqa: E501
 
     """
     Attributes:
@@ -48,7 +47,7 @@ class CustomerBillingType(object):
     }
 
     def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """CustomerBillingType - a model defined in OpenAPI"""  # noqa: E501
+        """ActorStatus - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
         self.local_vars_configuration = local_vars_configuration
@@ -88,14 +87,14 @@ class CustomerBillingType(object):
 
     def __eq__(self, other):
         """Returns true if both objects are equal"""
-        if not isinstance(other, CustomerBillingType):
+        if not isinstance(other, ActorStatus):
             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, CustomerBillingType):
+        if not isinstance(other, ActorStatus):
             return True
 
         return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/aggregated_instance_usage_csv.py

@@ -56,7 +56,10 @@ class AggregatedInstanceUsageCSV(object):
         'gpu_type': 'str',
         'memory_mib': 'int',
         'cpu_count': 'int',
-        'gpu_count': 'int'
+        'gpu_count': 'int',
+        'anyscale_managed_machine_pool_id': 'str',
+        'anyscale_managed_machine_pool_name': 'str',
+        'user_defined_machine_type': 'str'
     }
 
     attribute_map = {
@@ -83,10 +86,13 @@ class AggregatedInstanceUsageCSV(object):
         'gpu_type': 'gpu_type',
         'memory_mib': 'memory_mib',
         'cpu_count': 'cpu_count',
-        'gpu_count': 'gpu_count'
+        'gpu_count': 'gpu_count',
+        'anyscale_managed_machine_pool_id': 'anyscale_managed_machine_pool_id',
+        'anyscale_managed_machine_pool_name': 'anyscale_managed_machine_pool_name',
+        'user_defined_machine_type': 'user_defined_machine_type'
     }
 
-    def __init__(self, date=None, organization_id=None, organization_name=None, cloud_hosting_type=None, cloud_id=None, cloud_name=None, project_id=None, project_name=None, user_id=None, user_email=None, cluster_id=None, workspace_id=None, job_id=None, job_queue_id=None, service_id=None, instance_type=None, instance_seconds=None, cloud_provider=None, anyscale_credits=None, compute_stack=None, gpu_type=None, memory_mib=None, cpu_count=None, gpu_count=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, date=None, organization_id=None, organization_name=None, cloud_hosting_type=None, cloud_id=None, cloud_name=None, project_id=None, project_name=None, user_id=None, user_email=None, cluster_id=None, workspace_id=None, job_id=None, job_queue_id=None, service_id=None, instance_type=None, instance_seconds=None, cloud_provider=None, anyscale_credits=None, compute_stack=None, gpu_type=None, memory_mib=None, cpu_count=None, gpu_count=None, anyscale_managed_machine_pool_id=None, anyscale_managed_machine_pool_name=None, user_defined_machine_type=None, local_vars_configuration=None):  # noqa: E501
         """AggregatedInstanceUsageCSV - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -116,6 +122,9 @@ class AggregatedInstanceUsageCSV(object):
         self._memory_mib = None
         self._cpu_count = None
         self._gpu_count = None
+        self._anyscale_managed_machine_pool_id = None
+        self._anyscale_managed_machine_pool_name = None
+        self._user_defined_machine_type = None
         self.discriminator = None
 
         self.date = date
@@ -151,6 +160,12 @@ class AggregatedInstanceUsageCSV(object):
             self.cpu_count = cpu_count
         if gpu_count is not None:
             self.gpu_count = gpu_count
+        if anyscale_managed_machine_pool_id is not None:
+            self.anyscale_managed_machine_pool_id = anyscale_managed_machine_pool_id
+        if anyscale_managed_machine_pool_name is not None:
+            self.anyscale_managed_machine_pool_name = anyscale_managed_machine_pool_name
+        if user_defined_machine_type is not None:
+            self.user_defined_machine_type = user_defined_machine_type
 
     @property
     def date(self):
@@ -686,6 +701,69 @@ class AggregatedInstanceUsageCSV(object):
 
         self._gpu_count = gpu_count
 
+    @property
+    def anyscale_managed_machine_pool_id(self):
+        """Gets the anyscale_managed_machine_pool_id of this AggregatedInstanceUsageCSV.  # noqa: E501
+
+
+        :return: The anyscale_managed_machine_pool_id of this AggregatedInstanceUsageCSV.  # noqa: E501
+        :rtype: str
+        """
+        return self._anyscale_managed_machine_pool_id
+
+    @anyscale_managed_machine_pool_id.setter
+    def anyscale_managed_machine_pool_id(self, anyscale_managed_machine_pool_id):
+        """Sets the anyscale_managed_machine_pool_id of this AggregatedInstanceUsageCSV.
+
+
+        :param anyscale_managed_machine_pool_id: The anyscale_managed_machine_pool_id of this AggregatedInstanceUsageCSV.  # noqa: E501
+        :type: str
+        """
+
+        self._anyscale_managed_machine_pool_id = anyscale_managed_machine_pool_id
+
+    @property
+    def anyscale_managed_machine_pool_name(self):
+        """Gets the anyscale_managed_machine_pool_name of this AggregatedInstanceUsageCSV.  # noqa: E501
+
+
+        :return: The anyscale_managed_machine_pool_name of this AggregatedInstanceUsageCSV.  # noqa: E501
+        :rtype: str
+        """
+        return self._anyscale_managed_machine_pool_name
+
+    @anyscale_managed_machine_pool_name.setter
+    def anyscale_managed_machine_pool_name(self, anyscale_managed_machine_pool_name):
+        """Sets the anyscale_managed_machine_pool_name of this AggregatedInstanceUsageCSV.
+
+
+        :param anyscale_managed_machine_pool_name: The anyscale_managed_machine_pool_name of this AggregatedInstanceUsageCSV.  # noqa: E501
+        :type: str
+        """
+
+        self._anyscale_managed_machine_pool_name = anyscale_managed_machine_pool_name
+
+    @property
+    def user_defined_machine_type(self):
+        """Gets the user_defined_machine_type of this AggregatedInstanceUsageCSV.  # noqa: E501
+
+
+        :return: The user_defined_machine_type of this AggregatedInstanceUsageCSV.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_defined_machine_type
+
+    @user_defined_machine_type.setter
+    def user_defined_machine_type(self, user_defined_machine_type):
+        """Sets the user_defined_machine_type of this AggregatedInstanceUsageCSV.
+
+
+        :param user_defined_machine_type: The user_defined_machine_type of this AggregatedInstanceUsageCSV.  # noqa: E501
+        :type: str
+        """
+
+        self._user_defined_machine_type = user_defined_machine_type
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}