anyscale 0.26.4__py3-none-any.whl → 0.26.5__py3-none-any.whl

anyscale/client/openapi_client/models/operator_metrics.py

@@ -0,0 +1,256 @@
+# 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 OperatorMetrics(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',
+        'name': 'str',
+        'state': 'DatasetState',
+        'progress': 'int',
+        'total': 'int',
+        'metrics': 'dict(str, Metric)'
+    }
+
+    attribute_map = {
+        'id': 'id',
+        'name': 'name',
+        'state': 'state',
+        'progress': 'progress',
+        'total': 'total',
+        'metrics': 'metrics'
+    }
+
+    def __init__(self, id=None, name=None, state=None, progress=None, total=None, metrics=None, local_vars_configuration=None):  # noqa: E501
+        """OperatorMetrics - 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._name = None
+        self._state = None
+        self._progress = None
+        self._total = None
+        self._metrics = None
+        self.discriminator = None
+
+        self.id = id
+        self.name = name
+        self.state = state
+        self.progress = progress
+        self.total = total
+        self.metrics = metrics
+
+    @property
+    def id(self):
+        """Gets the id of this OperatorMetrics.  # noqa: E501
+
+
+        :return: The id of this OperatorMetrics.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id):
+        """Sets the id of this OperatorMetrics.
+
+
+        :param id: The id of this OperatorMetrics.  # 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 name(self):
+        """Gets the name of this OperatorMetrics.  # noqa: E501
+
+
+        :return: The name of this OperatorMetrics.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name):
+        """Sets the name of this OperatorMetrics.
+
+
+        :param name: The name of this OperatorMetrics.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and name is None:  # noqa: E501
+            raise ValueError("Invalid value for `name`, must not be `None`")  # noqa: E501
+
+        self._name = name
+
+    @property
+    def state(self):
+        """Gets the state of this OperatorMetrics.  # noqa: E501
+
+
+        :return: The state of this OperatorMetrics.  # noqa: E501
+        :rtype: DatasetState
+        """
+        return self._state
+
+    @state.setter
+    def state(self, state):
+        """Sets the state of this OperatorMetrics.
+
+
+        :param state: The state of this OperatorMetrics.  # noqa: E501
+        :type: DatasetState
+        """
+        if self.local_vars_configuration.client_side_validation and state is None:  # noqa: E501
+            raise ValueError("Invalid value for `state`, must not be `None`")  # noqa: E501
+
+        self._state = state
+
+    @property
+    def progress(self):
+        """Gets the progress of this OperatorMetrics.  # noqa: E501
+
+
+        :return: The progress of this OperatorMetrics.  # noqa: E501
+        :rtype: int
+        """
+        return self._progress
+
+    @progress.setter
+    def progress(self, progress):
+        """Sets the progress of this OperatorMetrics.
+
+
+        :param progress: The progress of this OperatorMetrics.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and progress is None:  # noqa: E501
+            raise ValueError("Invalid value for `progress`, must not be `None`")  # noqa: E501
+
+        self._progress = progress
+
+    @property
+    def total(self):
+        """Gets the total of this OperatorMetrics.  # noqa: E501
+
+
+        :return: The total of this OperatorMetrics.  # noqa: E501
+        :rtype: int
+        """
+        return self._total
+
+    @total.setter
+    def total(self, total):
+        """Sets the total of this OperatorMetrics.
+
+
+        :param total: The total of this OperatorMetrics.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and total is None:  # noqa: E501
+            raise ValueError("Invalid value for `total`, must not be `None`")  # noqa: E501
+
+        self._total = total
+
+    @property
+    def metrics(self):
+        """Gets the metrics of this OperatorMetrics.  # noqa: E501
+
+
+        :return: The metrics of this OperatorMetrics.  # noqa: E501
+        :rtype: dict(str, Metric)
+        """
+        return self._metrics
+
+    @metrics.setter
+    def metrics(self, metrics):
+        """Sets the metrics of this OperatorMetrics.
+
+
+        :param metrics: The metrics of this OperatorMetrics.  # noqa: E501
+        :type: dict(str, Metric)
+        """
+        if self.local_vars_configuration.client_side_validation and metrics is None:  # noqa: E501
+            raise ValueError("Invalid value for `metrics`, must not be `None`")  # noqa: E501
+
+        self._metrics = metrics
+
+    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, OperatorMetrics):
+            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, OperatorMetrics):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/train_run.py

@@ -42,6 +42,7 @@ class TrainRun(object):
         'start_time_ms': 'int',
         'end_time_ms': 'int',
         'attempts': 'list[TrainRunAttempt]',
+        'total_attempts': 'int',
         'job_details': 'JobDetails',
         'session_name': 'str',
         'controller_log_id': 'str'
@@ -57,12 +58,13 @@ class TrainRun(object):
         'start_time_ms': 'start_time_ms',
         'end_time_ms': 'end_time_ms',
         'attempts': 'attempts',
+        'total_attempts': 'total_attempts',
         'job_details': 'job_details',
         'session_name': 'session_name',
         'controller_log_id': 'controller_log_id'
     }
 
-    def __init__(self, id=None, name=None, job_id=None, controller_actor_id=None, status=None, status_detail=None, start_time_ms=None, end_time_ms=None, attempts=None, job_details=None, session_name=None, controller_log_id=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, id=None, name=None, job_id=None, controller_actor_id=None, status=None, status_detail=None, start_time_ms=None, end_time_ms=None, attempts=None, total_attempts=None, job_details=None, session_name=None, controller_log_id=None, local_vars_configuration=None):  # noqa: E501
         """TrainRun - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -77,6 +79,7 @@ class TrainRun(object):
         self._start_time_ms = None
         self._end_time_ms = None
         self._attempts = None
+        self._total_attempts = None
         self._job_details = None
         self._session_name = None
         self._controller_log_id = None
@@ -94,6 +97,7 @@ class TrainRun(object):
         if end_time_ms is not None:
             self.end_time_ms = end_time_ms
         self.attempts = attempts
+        self.total_attempts = total_attempts
         self.job_details = job_details
         if session_name is not None:
             self.session_name = session_name
@@ -301,6 +305,29 @@ class TrainRun(object):
 
         self._attempts = attempts
 
+    @property
+    def total_attempts(self):
+        """Gets the total_attempts of this TrainRun.  # noqa: E501
+
+
+        :return: The total_attempts of this TrainRun.  # noqa: E501
+        :rtype: int
+        """
+        return self._total_attempts
+
+    @total_attempts.setter
+    def total_attempts(self, total_attempts):
+        """Sets the total_attempts of this TrainRun.
+
+
+        :param total_attempts: The total_attempts of this TrainRun.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and total_attempts is None:  # noqa: E501
+            raise ValueError("Invalid value for `total_attempts`, must not be `None`")  # noqa: E501
+
+        self._total_attempts = total_attempts
+
     @property
     def job_details(self):
         """Gets the job_details of this TrainRun.  # noqa: E501

anyscale/cluster.py

@@ -121,10 +121,13 @@ def get_job_submission_client_cluster_info(
 
     if len(cluster_list) > 0:
         cluster = cluster_list[0]
-        if cluster.host_name and cluster.access_token:
+        access_token = api_client.get_cluster_access_token_api_v2_authentication_cluster_id_cluster_access_token_get(
+            cluster_id=cluster.id
+        )
+        if cluster.host_name and access_token:
             return ClusterInfo(
                 address=cluster.host_name,
-                cookies={"anyscale-token": cluster.access_token},
+                cookies={"anyscale-token": access_token},
                 metadata=metadata,
             )
         else:

anyscale/commands/anyscale_api/api_commands.py

@@ -4,7 +4,6 @@ Commands to interact with the Anyscale API
 
 import click
 
-from anyscale.commands.anyscale_api.session_commands_commands import session_commands
 from anyscale.commands.anyscale_api.session_operations_commands import (
     session_operations_commands,
 )
@@ -18,6 +17,5 @@ def anyscale_api() -> None:
     pass
 
 
-anyscale_api.add_command(session_commands)
 anyscale_api.add_command(sessions)
 anyscale_api.add_command(session_operations_commands)

anyscale/connect_utils/start_interactive_session.py

@@ -329,7 +329,10 @@ class StartInteractiveSessionBlock:
     ) -> Tuple[str, bool, Any]:
         """Get the params from the cluster needed to use Ray client."""
         connect_url = None
-        metadata = [("cookie", "anyscale-token=" + session_meta.access_token)]
+        access_token = self.api_client.get_cluster_access_token_api_v2_authentication_cluster_id_cluster_access_token_get(
+            cluster_id=session_meta.id
+        )
+        metadata = [("cookie", "anyscale-token=" + access_token)]
         if session_meta.connect_url:
             url_components = session_meta.connect_url.split("?port=")
             metadata += [("port", url_components[1])] if len(url_components) > 1 else []

anyscale/controllers/cluster_controller.py

@@ -358,22 +358,18 @@ class ClusterController(BaseController):
                         project_id = get_project_id(project_definition.root)
                     except click.ClickException:
                         parent_cloud_id = None
-                        feature_flag_on = self.api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
-                            "cloud-isolation-phase-1"
-                        ).result.is_on
                         log_msg = "No project context detected or `--project-id` provided. Continuing without a project."
-                        if feature_flag_on:
-                            if not (cloud_id or cloud_name):
-                                cloud_name = get_last_used_cloud(
-                                    project_id=None,
-                                    anyscale_api_client=self.anyscale_api_client,
-                                )
-                            parent_cloud_id, _ = get_cloud_id_and_name(
-                                api_client=self.api_client,
-                                cloud_id=cloud_id,
-                                cloud_name=cloud_name,
+                        if not (cloud_id or cloud_name):
+                            cloud_name = get_last_used_cloud(
+                                project_id=None,
+                                anyscale_api_client=self.anyscale_api_client,
                             )
-                            log_msg = f"No project context detected or `--project-id` provided. Continuing without a project assuming cloud {parent_cloud_id}."
+                        parent_cloud_id, _ = get_cloud_id_and_name(
+                            api_client=self.api_client,
+                            cloud_id=cloud_id,
+                            cloud_name=cloud_name,
+                        )
+                        log_msg = f"No project context detected or `--project-id` provided. Continuing without a project assuming cloud {parent_cloud_id}."
                         default_project = self.anyscale_api_client.get_default_project(
                             parent_cloud_id=parent_cloud_id
                         ).result
@@ -480,18 +476,11 @@ class ClusterController(BaseController):
                 "`--cloud-name` and `--cloud-id` cannot both be specified. Please only provide one "
                 "of these two arguments."
             )
-        feature_flag_on = self.api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
-            "cloud-isolation-phase-1"
-        ).result.is_on
-        if (
-            feature_flag_on
-            and cluster_name is not None
-            and (
-                project_id is None
-                and project_name is None
-                and cloud_id is None
-                and cloud_name is None
-            )
+        if cluster_name is not None and (
+            project_id is None
+            and project_name is None
+            and cloud_id is None
+            and cloud_name is None
         ):
             raise click.ClickException(
                 "Please specify the project of the cluster through `--project-id` or `--project-name`. If the cluster "

anyscale/controllers/project_controller.py

@@ -66,12 +66,7 @@ class ProjectController(BaseController):
         """
         Call API to create a new project given a project name.
         """
-        if not parent_cloud_id and (
-            self.api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
-                "cloud-isolation-phase-1"
-            ).result.is_on
-            is True
-        ):
+        if not parent_cloud_id:
             raise click.ClickException(
                 "Parent cloud id should be specified when creating a new project because "
                 "cloud isolation has been enabled."

anyscale/job/_private/job_sdk.py

@@ -295,7 +295,12 @@ class PrivateJobSDK(WorkloadSDK):
         runs = [self._job_run_model_to_job_run_status(run) for run in runs]
 
         return JobStatus(
-            name=model.name, id=model.id, state=state, runs=runs, config=config
+            name=model.name,
+            id=model.id,
+            state=state,
+            runs=runs,
+            config=config,
+            creator_id=model.creator_id,
         )
 
     def _resolve_to_job_model(

anyscale/job/models.py

@@ -487,6 +487,14 @@ state: STARTING
             if not isinstance(run, JobRunStatus):
                 raise TypeError("Each run in 'runs' must be a JobRunStatus.")
 
+    creator_id: str = field(
+        metadata={"docstring": "ID of the user who created the job.",},
+    )
+
+    def _validate_creator_id(self, creator_id: str):
+        if creator_id is not None and not isinstance(creator_id, str):
+            raise TypeError("'creator_id' must be a string.")
+
 
 class JobLogMode(ModelEnum):
     """Mode to use for getting job logs."""

anyscale/project_utils.py

@@ -277,10 +277,7 @@ def get_parent_cloud_id_and_name_of_project(
     project = api_client.get_project_api_v2_projects_project_id_get(
         project_id=project_id
     ).result
-    cloud_isolation_ff_on = api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
-        "cloud-isolation-phase-1"
-    ).result.is_on
-    if project and cloud_isolation_ff_on:
+    if project:
         parent_cloud_id, parent_cloud_name = get_cloud_id_and_name(
             api_client, cloud_id=project.parent_cloud_id
         )
@@ -439,22 +436,14 @@ def get_default_project(
         anyscale_api_client or get_auth_api_client(log_output=False).anyscale_api_client
     )
     api_client = api_client or get_auth_api_client(log_output=False).api_client
-    if (
-        api_client.check_is_feature_flag_on_api_v2_userinfo_check_is_feature_flag_on_get(
-            "cloud-isolation-phase-1"
-        ).result.is_on
-        is False
-    ):
-        default_project = anyscale_api_client.get_default_project().result
-    else:
-        if not parent_cloud_id:
-            raise click.ClickException(
-                "Please specify a cloud for this command either through the cloud or "
-                "compute config arguments."
-            )
-        default_project = anyscale_api_client.get_default_project(
-            parent_cloud_id=parent_cloud_id
-        ).result
+    if not parent_cloud_id:
+        raise click.ClickException(
+            "Please specify a cloud for this command either through the cloud or "
+            "compute config arguments."
+        )
+    default_project = anyscale_api_client.get_default_project(
+        parent_cloud_id=parent_cloud_id
+    ).result
     return default_project
 
 

anyscale/sdk/anyscale_client/__init__.py

@@ -113,7 +113,6 @@ from anyscale_client.models.create_project import CreateProject
 from anyscale_client.models.create_sso_config import CreateSSOConfig
 from anyscale_client.models.create_schedule import CreateSchedule
 from anyscale_client.models.create_session import CreateSession
-from anyscale_client.models.create_session_command import CreateSessionCommand
 from anyscale_client.models.grpc_protocol_config import GrpcProtocolConfig
 from anyscale_client.models.http_validation_error import HTTPValidationError
 from anyscale_client.models.ha_job_goal_states import HaJobGoalStates
@@ -188,8 +187,6 @@ from anyscale_client.models.service_version_state import ServiceVersionState
 from anyscale_client.models.servicemodel_list_response import ServicemodelListResponse
 from anyscale_client.models.servicemodel_response import ServicemodelResponse
 from anyscale_client.models.session import Session
-from anyscale_client.models.session_command import SessionCommand
-from anyscale_client.models.session_command_types import SessionCommandTypes
 from anyscale_client.models.session_list_response import SessionListResponse
 from anyscale_client.models.session_operation import SessionOperation
 from anyscale_client.models.session_operation_type import SessionOperationType
@@ -198,8 +195,6 @@ from anyscale_client.models.session_starting_up_data import SessionStartingUpDat
 from anyscale_client.models.session_state import SessionState
 from anyscale_client.models.session_state_data import SessionStateData
 from anyscale_client.models.session_stopping_data import SessionStoppingData
-from anyscale_client.models.sessioncommand_list_response import SessioncommandListResponse
-from anyscale_client.models.sessioncommand_response import SessioncommandResponse
 from anyscale_client.models.sessionoperation_response import SessionoperationResponse
 from anyscale_client.models.sessions_query import SessionsQuery
 from anyscale_client.models.sort_by_clause_jobs_sort_field import SortByClauseJobsSortField