anyscale 0.26.15__py3-none-any.whl → 0.26.17__py3-none-any.whl

anyscale/schedule/commands.py

@@ -36,12 +36,14 @@ _APPLY_ARG_DOCSTRINGS = {"config": "The config options defining the schedule."}
     doc_py_example=_APPLY_EXAMPLE,
     arg_docstrings=_APPLY_ARG_DOCSTRINGS,
 )
-def apply(config: ScheduleConfig, *, _sdk: PrivateScheduleSDK) -> str:
+def apply(
+    config: ScheduleConfig, *, _private_sdk: Optional[PrivateScheduleSDK] = None
+) -> str:
     """Apply or update a schedule.
 
     Returns the id of the schedule.
     """
-    return _sdk.apply(config)
+    return _private_sdk.apply(config)  # type: ignore
 
 
 _SET_STATE_EXAMPLE = """
@@ -76,13 +78,15 @@ def set_state(
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     state: ScheduleState,
-    _sdk: PrivateScheduleSDK
+    _private_sdk: Optional[PrivateScheduleSDK] = None
 ) -> str:
     """Set the state of a schedule.
 
     Returns the id of the schedule.
     """
-    return _sdk.set_state(id=id, name=name, cloud=cloud, project=project, state=state,)
+    return _private_sdk.set_state(  # type: ignore
+        id=id, name=name, cloud=cloud, project=project, state=state,
+    )
 
 
 _STATUS_EXAMPLE = """
@@ -110,11 +114,11 @@ def status(
     name: Optional[str] = None,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateScheduleSDK
+    _private_sdk: Optional[PrivateScheduleSDK] = None
 ) -> ScheduleStatus:
     """Return the status of the schedule.
     """
-    return _sdk.status(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.status(id=id, name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _TRIGGER_EXAMPLE = """
@@ -142,8 +146,8 @@ def trigger(
     name: Optional[str] = None,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateScheduleSDK
+    _private_sdk: Optional[PrivateScheduleSDK] = None
 ) -> str:
     """Trigger the execution of the schedule.
     """
-    return _sdk.trigger(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.trigger(id=id, name=name, cloud=cloud, project=project)  # type: ignore

anyscale/sdk/anyscale_client/sdk.py

@@ -477,6 +477,7 @@ class AnyscaleSDK(DefaultApi):  # type: ignore
             create_cluster_environment, poll_rate_seconds, timeout_seconds
         )
 
+        assert cluster_environment_build is not None
         return self.launch_cluster(
             project_id,
             cluster_name,

anyscale/service/commands.py

@@ -48,7 +48,7 @@ def deploy(
     in_place: bool = False,
     canary_percent: Optional[int] = None,
     max_surge_percent: Optional[int] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Deploy a service.
 
@@ -58,7 +58,7 @@ def deploy(
 
     Returns the id of the deployed service.
     """
-    return _sdk.deploy(
+    return _private_sdk.deploy(  # type: ignore
         config,
         in_place=in_place,
         canary_percent=canary_percent,
@@ -92,7 +92,7 @@ def rollback(
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     max_surge_percent: Optional[int] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Rollback to the primary version of the service.
 
@@ -103,7 +103,7 @@ def rollback(
 
     Returns the id of the rolled back service.
     """
-    return _sdk.rollback(
+    return _private_sdk.rollback(  # type: ignore
         name=name, cloud=cloud, project=project, max_surge_percent=max_surge_percent
     )
 
@@ -132,7 +132,7 @@ def terminate(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Terminate a service.
 
@@ -140,7 +140,7 @@ def terminate(
 
     Returns the id of the terminated service.
     """
-    return _sdk.terminate(name=name, cloud=cloud, project=project)
+    return _private_sdk.terminate(name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _ARCHIVE_EXAMPLE = """
@@ -169,7 +169,7 @@ def archive(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Archive a service.
 
@@ -177,7 +177,7 @@ def archive(
 
     Returns the ID of the archived service.
     """
-    return _sdk.archive(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.archive(id=id, name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _DELETE_EXAMPLE = """
@@ -206,7 +206,7 @@ def delete(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> str:
     """Delete a service.
 
@@ -214,7 +214,7 @@ def delete(
 
     Returns the ID of the deleted service.
     """
-    return _sdk.delete(id=id, name=name, cloud=cloud, project=project)
+    return _private_sdk.delete(id=id, name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _STATUS_EXAMPLE = """
@@ -242,10 +242,10 @@ def status(
     *,
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ) -> ServiceStatus:
     """Get the status of a service."""
-    return _sdk.status(name=name, cloud=cloud, project=project)
+    return _private_sdk.status(name=name, cloud=cloud, project=project)  # type: ignore
 
 
 _WAIT_EXAMPLE = """
@@ -277,18 +277,18 @@ def wait(
     project: Optional[str] = None,
     state: Union[str, ServiceState] = ServiceState.RUNNING,
     timeout_s: float = 600,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
     _interval_s: float = 5,
 ):
     """Wait for a service to reach a target state."""
-    _sdk.wait(
+    _private_sdk.wait(  # type: ignore
         name=name,
         cloud=cloud,
         project=project,
         state=ServiceState(state),
         timeout_s=timeout_s,
         interval_s=_interval_s,
-    )
+    )  # type: ignore
 
 
 _CONTROLLER_LOGS_EXAMPLE = """
@@ -322,10 +322,10 @@ def _controller_logs(
     canary: bool = False,
     mode: Union[str, ServiceLogMode] = ServiceLogMode.TAIL,
     max_lines: int = 1000,
-    _sdk: PrivateServiceSDK,
+    _private_sdk: Optional[PrivateServiceSDK] = None,
 ):
     """Wait for a service to reach a target state."""
-    return _sdk.controller_logs(
+    return _private_sdk.controller_logs(  # type: ignore
         name,
         cloud=cloud,
         project=project,

anyscale/service_account/commands.py

@@ -77,10 +77,12 @@ _ROTATE_API_KEYS_DOCSTRINGS = {
     doc_py_example=_CREATE_EXAMPLE,
     arg_docstrings=_CREATE_DOCSTRINGS,
 )
-def create(name: str, *, _sdk: PrivateServiceAccountSDK) -> str:
+def create(
+    name: str, *, _private_sdk: Optional[PrivateServiceAccountSDK] = None
+) -> str:
     """Create a service account and return the API key.
     """
-    return _sdk.create(name)
+    return _private_sdk.create(name)  # type: ignore
 
 
 @sdk_command(
@@ -93,11 +95,11 @@ def create_api_key(
     email: Optional[str] = None,
     name: Optional[str] = None,
     *,
-    _sdk: PrivateServiceAccountSDK
+    _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ) -> str:
     """Create an API key for the service account and return the API key.
     """
-    return _sdk.create_api_key(email, name)
+    return _private_sdk.create_api_key(email, name)  # type: ignore
 
 
 @sdk_command(
@@ -107,10 +109,10 @@ def create_api_key(
     arg_docstrings=_LIST_DOCSTRINGS,
 )
 def list(  # noqa: A001
-    max_items: int = 20, *, _sdk: PrivateServiceAccountSDK
+    max_items: int = 20, *, _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ) -> List[ServiceAccount]:
     """List service accounts. """
-    return _sdk.list(max_items)
+    return _private_sdk.list(max_items)  # type: ignore
 
 
 @sdk_command(
@@ -123,11 +125,11 @@ def delete(
     email: Optional[str] = None,
     name: Optional[str] = None,
     *,
-    _sdk: PrivateServiceAccountSDK
+    _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ):
     """Delete a service account.
     """
-    return _sdk.delete(email, name)
+    return _private_sdk.delete(email, name)  # type: ignore
 
 
 @sdk_command(
@@ -140,8 +142,8 @@ def rotate_api_keys(
     email: Optional[str] = None,
     name: Optional[str] = None,
     *,
-    _sdk: PrivateServiceAccountSDK
+    _private_sdk: Optional[PrivateServiceAccountSDK] = None
 ) -> str:
     """Rotate all api keys of a service account and return the new API key.
     """
-    return _sdk.rotate_api_keys(email, name)
+    return _private_sdk.rotate_api_keys(email, name)  # type: ignore

anyscale/user/commands.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from anyscale._private.sdk import sdk_command
 from anyscale.user._private.user_sdk import PrivateUserSDK
@@ -35,8 +35,10 @@ _ADMIN_BATCH_CREATE_ARG_DOCSTRINGS = {
     arg_docstrings=_ADMIN_BATCH_CREATE_ARG_DOCSTRINGS,
 )
 def admin_batch_create(
-    admin_create_users: List[AdminCreateUser], *, _sdk: PrivateUserSDK
+    admin_create_users: List[AdminCreateUser],
+    *,
+    _private_sdk: Optional[PrivateUserSDK] = None
 ) -> List[AdminCreatedUser]:
     """Batch create, as an admin, users without email verification.
     """
-    return _sdk.admin_batch_create(admin_create_users)
+    return _private_sdk.admin_batch_create(admin_create_users)  # type: ignore

anyscale/utils/gcp_utils.py

@@ -41,15 +41,31 @@ GCP_REQUIRED_APIS = [
 ]
 
 GCP_RESOURCE_DICT: Dict[str, CloudAnalyticsEventCloudResource] = {
-    "VPC": CloudAnalyticsEventCloudResource.GCP_VPC,
-    "Subnet": CloudAnalyticsEventCloudResource.GCP_SUBNET,
-    "Project": CloudAnalyticsEventCloudResource.GCP_PROJECT,
-    "Anyscale Access Service Account": CloudAnalyticsEventCloudResource.GCP_SERVICE_ACCOUNT,
-    "Dataplane Service Account": CloudAnalyticsEventCloudResource.GCP_SERVICE_ACCOUNT,
-    "Firewall Policy": CloudAnalyticsEventCloudResource.GCP_FIREWALL_POLICY,
-    "Filestore": CloudAnalyticsEventCloudResource.GCP_FILESTORE,
-    "Google Cloud Storage Bucket": CloudAnalyticsEventCloudResource.GCP_STORAGE_BUCKET,
-    "Memorystore": CloudAnalyticsEventCloudResource.GCP_MEMORYSTORE,
+    "VPC": CloudAnalyticsEventCloudResource(CloudAnalyticsEventCloudResource.GCP_VPC),
+    "Subnet": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_SUBNET
+    ),
+    "Project": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_PROJECT
+    ),
+    "Anyscale Access Service Account": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_SERVICE_ACCOUNT
+    ),
+    "Dataplane Service Account": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_SERVICE_ACCOUNT
+    ),
+    "Firewall Policy": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_FIREWALL_POLICY
+    ),
+    "Filestore": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_FILESTORE
+    ),
+    "Google Cloud Storage Bucket": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_STORAGE_BUCKET
+    ),
+    "Memorystore": CloudAnalyticsEventCloudResource(
+        CloudAnalyticsEventCloudResource.GCP_MEMORYSTORE
+    ),
 }
 
 

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.26.15"
+__version__ = "0.26.17"

anyscale/workspace/__init__.py

@@ -253,7 +253,7 @@ class WorkspaceSDK:
         doc_py_example=_UPDATE_EXAMPLE, arg_docstrings=_UPDATE_ARG_DOCSTRINGS,
     )
     def update(  # noqa: F811
-        self, *, id: str = None, config: UpdateWorkspaceConfig  # noqa: A002
+        self, *, id: Optional[str] = None, config: UpdateWorkspaceConfig  # noqa: A002
     ):
         """Update a workspace."""
         self._private_sdk.update(

anyscale/workspace/_private/workspace_sdk.py

@@ -292,7 +292,7 @@ class PrivateWorkspaceSDK(WorkloadSDK):
                 f"Workspace '{workspace_id_or_name}' did not reach target state {state} within {timeout_s}s. Last seen state: {curr_state}."
             )
 
-    def _get_workspace_status(self, workspace_id: str) -> WorkspaceState:
+    def _get_workspace_status(self, workspace_id: Optional[str]) -> WorkspaceState:
 
         cluster = self.client.get_workspace_cluster(workspace_id)
         if not cluster:
@@ -564,7 +564,9 @@ class PrivateWorkspaceSDK(WorkloadSDK):
             delete=delete,
         )
 
-    def update(self, *, id: str, config: UpdateWorkspaceConfig) -> str:  # noqa: A002
+    def update(
+        self, *, id: Optional[str], config: UpdateWorkspaceConfig  # noqa: A002
+    ) -> str:
         workspace = self.client.get_workspace(id=id)  # type: ignore
 
         if not workspace:
@@ -642,7 +644,7 @@ class PrivateWorkspaceSDK(WorkloadSDK):
             )
             self._logger.info(f"Applied environment variables to workspace id: {id}")
 
-        return id
+        return id  # type: ignore
 
     def _convert_cluster_state_to_workspace_state(
         self, state: SessionState

anyscale/workspace/commands.py

@@ -34,12 +34,14 @@ _WAIT_TIMEOUT_SECONDS = 1800.0
     doc_py_example=_CREATE_EXAMPLE,
     arg_docstrings=_CREATE_ARG_DOCSTRINGS,
 )
-def create(config: WorkspaceConfig, *, _sdk: PrivateWorkspaceSDK) -> str:
+def create(
+    config: WorkspaceConfig, *, _private_sdk: Optional[PrivateWorkspaceSDK] = None
+) -> str:
     """Create a workspace.
 
     Returns the id of the created workspace.
     """
-    return _sdk.create(config)
+    return _private_sdk.create(config)  # type: ignore
 
 
 _START_EXAMPLE = """
@@ -70,13 +72,13 @@ def start(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> str:
     """Start a workspace.
 
     Returns the id of the started workspace.
     """
-    return _sdk.start(name=name, id=id, cloud=cloud, project=project)
+    return _private_sdk.start(name=name, id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _TERMINATE_EXAMPLE = """
@@ -107,13 +109,13 @@ def terminate(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> str:
     """Terminate a workspace.
 
     Returns the id of the terminated workspace.
     """
-    return _sdk.terminate(name=name, id=id, cloud=cloud, project=project)
+    return _private_sdk.terminate(name=name, id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _STATUS_EXAMPLE = """
@@ -144,13 +146,13 @@ def status(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> str:
     """Get the status of a workspace.
 
     Returns the status of the workspace.
     """
-    return _sdk.status(name=name, id=id, cloud=cloud, project=project)
+    return _private_sdk.status(name=name, id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _WAIT_EXAMPLE = """
@@ -185,13 +187,13 @@ def wait(
     project: Optional[str] = None,
     timeout_s: float = _WAIT_TIMEOUT_SECONDS,
     state: str = WorkspaceState.RUNNING,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> str:
     """Wait for a workspace to reach a terminal state.
 
     Returns the status of the workspace.
     """
-    return _sdk.wait(
+    return _private_sdk.wait(  # type: ignore
         name=name,
         id=id,
         cloud=cloud,
@@ -235,13 +237,13 @@ def generate_ssh_config_file(
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     ssh_config_path: Optional[str] = None,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> Tuple[str, str]:
     """Generate an SSH config file for a workspace.
 
     Returns the hostname and path to the generated config file.
     """
-    return _sdk.generate_ssh_config_file(
+    return _private_sdk.generate_ssh_config_file(  # type: ignore
         name=name, id=id, cloud=cloud, project=project, ssh_config_path=ssh_config_path,
     )
 
@@ -281,14 +283,14 @@ def run_command(
     cloud: Optional[str] = None,
     project: Optional[str] = None,
     command: str,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
     **kwargs: Dict[str, Any],
 ):
     """Run a command in a workspace.
 
     Returns a subprocess.CompletedProcess object.
     """
-    return _sdk.run_command(
+    return _private_sdk.run_command(  # type: ignore
         name=name, id=id, cloud=cloud, project=project, command=command, **kwargs,
     )
 
@@ -328,13 +330,13 @@ def pull(
     pull_git_state: bool = False,
     rsync_args: Optional[List[str]] = None,
     delete: bool = False,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> None:
     """Pull a workspace to a local directory.
 
     Returns the path to the pulled workspace.
     """
-    _sdk.pull(
+    _private_sdk.pull(  # type: ignore
         name=name,
         id=id,
         cloud=cloud,
@@ -383,13 +385,13 @@ def push(
     push_git_state: bool = False,
     rsync_args: Optional[List[str]] = None,
     delete: bool = False,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> None:
     """Push a local directory to a workspace.
 
     Returns the path to the pushed workspace.
     """
-    _sdk.push(
+    _private_sdk.push(  # type: ignore
         name=name,
         id=id,
         cloud=cloud,
@@ -427,13 +429,13 @@ _UPDATE_ARG_DOCSTRINGS = {
 )
 def update(
     *,
-    id: str = None,  # noqa: A002
+    id: Optional[str] = None,  # noqa: A002
     config: UpdateWorkspaceConfig,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> None:
     """Update a workspace."""
-    _sdk.update(
-        id=id, config=config,  # type: ignore
+    _private_sdk.update(  # type: ignore
+        id=id, config=config,
     )
 
 
@@ -466,7 +468,7 @@ def get(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateWorkspaceSDK,
+    _private_sdk: Optional[PrivateWorkspaceSDK] = None,
 ) -> Workspace:
     """Get a workspace."""
-    return _sdk.get(name=name, id=id, cloud=cloud, project=project)
+    return _private_sdk.get(name=name, id=id, cloud=cloud, project=project)  # type: ignore

{anyscale-0.26.15.dist-info → anyscale-0.26.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.26.15
+Version: 0.26.17
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License