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

anyscale/controllers/cloud_controller.py

@@ -3,6 +3,7 @@ Fetches data required and formats output for `anyscale cloud` commands.
 """
 
 import copy
+from datetime import datetime, timedelta
 import json
 from os import getenv
 import pathlib
@@ -16,6 +17,7 @@ import boto3
 from botocore.exceptions import ClientError, NoCredentialsError
 import click
 from click import Abort, ClickException
+from rich.progress import Progress, track
 import yaml
 
 from anyscale import __version__ as anyscale_version
@@ -72,8 +74,12 @@ from anyscale.controllers.cloud_functional_verification_controller import (
     CloudFunctionalVerificationType,
 )
 from anyscale.formatters import clouds_formatter
+from anyscale.job._private.job_sdk import (
+    HA_JOB_STATE_TO_JOB_STATE,
+    TERMINAL_HA_JOB_STATES,
+)
 from anyscale.shared_anyscale_utils.aws import AwsRoleArn
-from anyscale.shared_anyscale_utils.conf import ANYSCALE_ENV
+from anyscale.shared_anyscale_utils.conf import ANYSCALE_ENV, ANYSCALE_HOST
 from anyscale.util import (  # pylint:disable=private-import
     _client,
     _get_aws_efs_mount_target_ip,
@@ -1411,7 +1417,7 @@ class CloudController(BaseController):
         )
         if enable_log_ingestion is not None:
             self._update_customer_aggregated_logs_config(
-                cloud_id=cloud_id, is_enabled=enable_log_ingestion,
+                cloud_id=cloud_id, is_enabled=enable_log_ingestion,  # type: ignore
             )
             self.log.info(
                 f"Successfully updated log ingestion configuration for cloud, "
@@ -1552,7 +1558,7 @@ class CloudController(BaseController):
                 cloud_resource=cloud_resource,
                 boto3_session=boto3_session,
                 region=cloud.region,
-                cloud_id=cloud_id,
+                cloud_id=cloud_id,  # type: ignore
                 is_bring_your_own_resource=cloud.is_bring_your_own_resource,
                 is_private_network=cloud.is_private_cloud
                 if cloud.is_private_cloud
@@ -1575,7 +1581,7 @@ class CloudController(BaseController):
                 project_id=project_id,
                 host_project_id=host_project_id,
                 region=cloud.region,
-                cloud_id=cloud_id,
+                cloud_id=cloud_id,  # type: ignore
                 yes=False,
                 strict=strict,
                 is_private_service_cloud=cloud.is_private_service_cloud,
@@ -3539,8 +3545,8 @@ class CloudController(BaseController):
                     )
                 try:
                     self._edit_aws_cloud(
-                        cloud_id=cloud_id,
-                        cloud_name=cloud_name,
+                        cloud_id=cloud_id,  # type: ignore
+                        cloud_name=cloud_name,  # type: ignore
                         cloud=cloud,
                         cloud_resource=cloud_resource,
                         aws_s3_id=aws_s3_id,
@@ -3593,8 +3599,8 @@ class CloudController(BaseController):
                     )
                 try:
                     self._edit_gcp_cloud(
-                        cloud_id=cloud_id,
-                        cloud_name=cloud_name,
+                        cloud_id=cloud_id,  # type: ignore
+                        cloud_name=cloud_name,  # type: ignore
                         cloud=cloud,
                         cloud_resource=cloud_resource,
                         gcp_filestore_instance_id=gcp_filestore_instance_id,
@@ -3643,3 +3649,140 @@ class CloudController(BaseController):
                 )
 
     ### End of edit cloud ###
+
+    def generate_jobs_report(self, cloud_id: str, csv: bool, out_path: str) -> None:
+        end_time = datetime.now()
+        start_time = end_time - timedelta(days=7)
+
+        full_results = []
+        paging_token: Optional[str] = None
+        count_per_page = 20
+        curr_page_results = None
+        total_jobs: Optional[int] = None
+
+        with Progress() as progress:
+            download_task = progress.add_task("Downloading jobs...", total=None)
+
+            while curr_page_results is None or len(curr_page_results) == count_per_page:
+                response = self.api_client.list_job_reports_api_v2_job_reports_get(
+                    cloud_id,
+                    start_time=start_time,
+                    end_time=end_time,
+                    paging_token=paging_token,
+                    count=count_per_page,
+                )
+                curr_page_results = response.results
+                full_results.extend(curr_page_results)
+                paging_token = response.metadata.next_paging_token
+                total_jobs = response.metadata.total
+                progress.update(
+                    download_task, total=total_jobs, advance=len(curr_page_results)
+                )
+
+            progress.update(download_task, completed=total_jobs)
+
+        if not full_results:
+            self.log.info("No jobs found in the last 7 days.")
+            return
+
+        filtered_results = [
+            job
+            for job in full_results
+            if job.job_state in TERMINAL_HA_JOB_STATES and job.job_report is not None
+        ]
+        filtered_results.sort(key=lambda x: x.created_at)
+
+        with open(out_path, "w") as out_file:
+            if csv:
+                out_file.write(
+                    "Job ID,Job name,Job state,Created at,Finished at,Duration,Unused CPU hours,Unused GPU hours,Max concurrent instances\n"
+                )
+                for job in track(filtered_results, description="Generating report..."):
+                    job_state = HA_JOB_STATE_TO_JOB_STATE[job.job_state]
+                    if job.finished_at is not None:
+                        duration = str(job.finished_at - job.created_at)
+                        finished_at = str(job.finished_at)
+                    else:
+                        duration = ""
+                        finished_at = ""
+                    unused_cpu_hours = job.job_report.unused_cpu_hours or ""
+                    unused_gpu_hours = job.job_report.unused_gpu_hours or ""
+                    max_instances_launched = job.job_report.max_instances_launched or ""
+
+                    out_file.write(
+                        f"{job.job_id},{job.job_name},{job_state},{str(job.created_at)},{finished_at},{duration},{unused_cpu_hours},{unused_gpu_hours},{max_instances_launched}\n"
+                    )
+            else:
+                out_file.write(
+                    f"""
+<html>
+<head>
+<title>Jobs Report - {str(end_time)}</title>
+<style>
+    table {{
+        border: 1px solid black;
+        border-collapse: collapse;
+    }}
+    th, td {{
+        border: 1px solid black;
+        border-collapse: collapse;
+        padding: 8px;
+    }}
+</style>
+</head>
+<body>
+<h1>Job Report - {str(end_time)}</h1>
+<p>Total jobs reported (finished jobs): {len(filtered_results)}</p>
+<p>Total jobs in the last 7 days: {total_jobs}</p>
+<table>
+<thead>
+<tr>
+<th>Job ID</th>
+<th>Job name</th>
+<th>Job state</th>
+<th>Created at</th>
+<th>Finished at</th>
+<th>Duration</th>
+<th>Unused CPU hours</th>
+<th>Unused GPU hours</th>
+<th>Max concurrent instances</th>
+</tr>
+</thead>
+<tbody>
+"""
+                )
+
+                for job in track(filtered_results, description="Generating report..."):
+                    job_state = HA_JOB_STATE_TO_JOB_STATE[job.job_state]
+                    if job.finished_at is not None:
+                        duration = str(job.finished_at - job.created_at)
+                        finished_at = str(job.finished_at)
+                    else:
+                        duration = ""
+                        finished_at = ""
+                    unused_cpu_hours = job.job_report.unused_cpu_hours or ""
+                    unused_gpu_hours = job.job_report.unused_gpu_hours or ""
+                    max_instances_launched = job.job_report.max_instances_launched or ""
+
+                    out_file.write(
+                        f"""
+<tr>
+<td><a target="_blank" rel="noreferrer" href="{ANYSCALE_HOST}/jobs/{job.job_id}">{job.job_id}</a></td>
+<td>{job.job_name}</td>
+<td>{job_state}</td>
+<td>{str(job.created_at)}</td>
+<td>{finished_at}</td>
+<td>{duration}</td>
+<td>{unused_cpu_hours}</td>
+<td>{unused_gpu_hours}</td>
+<td>{max_instances_launched}</td>
+</tr>
+"""
+                    )
+
+                out_file.write(
+                    """
+</tbody>
+</table>
+"""
+                )

anyscale/controllers/cluster_controller.py

@@ -522,6 +522,7 @@ class ClusterController(BaseController):
                 api_client=self.api_client,
                 anyscale_api_client=self.anyscale_api_client,
             )
+            assert project_id is not None
             cluster_name = self._get_or_generate_cluster_name(project_id, cluster_name)
         return project_id, cluster_name
 

anyscale/controllers/job_controller.py

@@ -80,7 +80,7 @@ class JobController(BaseController):
         log: Optional[LogsLogger] = None,
         initialize_auth_api_client: bool = True,
         raise_structured_exception: bool = False,
-        auth_token: str = None,
+        auth_token: Optional[str] = None,
     ):
         if log is None:
             log = LogsLogger()

anyscale/controllers/service_controller.py

@@ -354,6 +354,7 @@ class ServiceController(BaseController):
                 )
 
         else:
+            assert ray_serve_config is not None
             ray_serve_config["runtime_env"] = override_runtime_env_config(
                 runtime_env=ray_serve_config.get("runtime_env"),
                 anyscale_api_client=self.sdk,

anyscale/image/commands.py

@@ -36,13 +36,13 @@ def build(
     *,
     name: str,
     ray_version: Optional[str] = None,
-    _sdk: PrivateImageSDK
+    _private_sdk: Optional[PrivateImageSDK] = None,
 ) -> str:
     """Build an image from a Containerfile.
 
     Returns the URI of the image.
     """
-    return _sdk.build_image_from_containerfile_with_image_uri(
+    return _private_sdk.build_image_from_containerfile_with_image_uri(  # type: ignore
         name, containerfile, ray_version=ray_version
     )
 
@@ -68,12 +68,12 @@ _GET_ARG_DOCSTRINGS = {
     doc_py_example=_GET_EXAMPLE,
     arg_docstrings=_GET_ARG_DOCSTRINGS,
 )
-def get(*, name: str, _sdk: PrivateImageSDK) -> ImageBuild:
+def get(*, name: str, _private_sdk: Optional[PrivateImageSDK] = None) -> ImageBuild:
     """The name can contain an optional version tag, i.e., 'name:version'.
 
     If no version is provided, the latest one will be returned.
     """
-    return _sdk.get(name)
+    return _private_sdk.get(name)  # type: ignore
 
 
 _REGISTER_EXAMPLE = """
@@ -102,14 +102,14 @@ def register(
     name: str,
     ray_version: Optional[str] = None,
     registry_login_secret: Optional[str] = None,
-    _sdk: PrivateImageSDK
+    _private_sdk: Optional[PrivateImageSDK] = None,
 ) -> str:
     """
     Registers a BYOD image with a container image name.
 
     Returns the URI of the image.
     """
-    return _sdk.register_byod_image_with_name(
+    return _private_sdk.register_byod_image_with_name(  # type: ignore
         image_uri,
         registry_login_secret=registry_login_secret,
         ray_version=ray_version,

anyscale/job/_private/job_sdk.py

@@ -33,6 +33,26 @@ from anyscale.utils.runtime_env import parse_requirements_file
 
 logger = BlockLogger()
 
+HA_JOB_STATE_TO_JOB_STATE = {
+    HaJobStates.UPDATING: JobState.RUNNING,
+    HaJobStates.RUNNING: JobState.RUNNING,
+    HaJobStates.RESTARTING: JobState.RUNNING,
+    HaJobStates.CLEANING_UP: JobState.RUNNING,
+    HaJobStates.PENDING: JobState.STARTING,
+    HaJobStates.AWAITING_CLUSTER_START: JobState.STARTING,
+    HaJobStates.SUCCESS: JobState.SUCCEEDED,
+    HaJobStates.ERRORED: JobState.FAILED,
+    HaJobStates.TERMINATED: JobState.FAILED,
+    HaJobStates.BROKEN: JobState.FAILED,
+    HaJobStates.OUT_OF_RETRIES: JobState.FAILED,
+}
+
+TERMINAL_HA_JOB_STATES = [
+    HaJobStates.SUCCESS,
+    HaJobStates.TERMINATED,
+    HaJobStates.OUT_OF_RETRIES,
+]
+
 
 class PrivateJobSDK(WorkloadSDK):
     _POLLING_INTERVAL_SECONDS = 10.0
@@ -215,20 +235,6 @@ class PrivateJobSDK(WorkloadSDK):
         )
         return job.id
 
-    _HA_JOB_STATE_TO_JOB_STATE = {
-        HaJobStates.UPDATING: JobState.RUNNING,
-        HaJobStates.RUNNING: JobState.RUNNING,
-        HaJobStates.RESTARTING: JobState.RUNNING,
-        HaJobStates.CLEANING_UP: JobState.RUNNING,
-        HaJobStates.PENDING: JobState.STARTING,
-        HaJobStates.AWAITING_CLUSTER_START: JobState.STARTING,
-        HaJobStates.SUCCESS: JobState.SUCCEEDED,
-        HaJobStates.ERRORED: JobState.FAILED,
-        HaJobStates.TERMINATED: JobState.FAILED,
-        HaJobStates.BROKEN: JobState.FAILED,
-        HaJobStates.OUT_OF_RETRIES: JobState.FAILED,
-    }
-
     _BACKEND_JOB_STATUS_TO_JOB_RUN_STATE = {
         BackendJobStatus.RUNNING: JobRunState.RUNNING,
         BackendJobStatus.COMPLETED: JobRunState.SUCCEEDED,
@@ -241,9 +247,7 @@ class PrivateJobSDK(WorkloadSDK):
 
     def _job_state_from_job_model(self, model: ProductionJob) -> JobState:
         ha_state = model.state.current_state if model.state else None
-        return cast(
-            JobState, self._HA_JOB_STATE_TO_JOB_STATE.get(ha_state, JobState.UNKNOWN)
-        )
+        return cast(JobState, HA_JOB_STATE_TO_JOB_STATE.get(ha_state, JobState.UNKNOWN))
 
     def _job_run_model_to_job_run_status(self, run: Job) -> JobRunStatus:
         state = self._BACKEND_JOB_STATUS_TO_JOB_RUN_STATE.get(
@@ -358,12 +362,6 @@ class PrivateJobSDK(WorkloadSDK):
         self.logger.info(f"Marked job '{job_model.name}' for termination")
         return job_model.id
 
-    _TERMINAL_HA_JOB_STATES = [
-        HaJobStates.SUCCESS,
-        HaJobStates.TERMINATED,
-        HaJobStates.OUT_OF_RETRIES,
-    ]
-
     def archive(
         self,
         *,
@@ -377,7 +375,7 @@ class PrivateJobSDK(WorkloadSDK):
         )
 
         ha_state = job_model.state.current_state if job_model.state else None
-        if ha_state not in self._TERMINAL_HA_JOB_STATES:
+        if ha_state not in TERMINAL_HA_JOB_STATES:
             raise RuntimeError(
                 f"Job with ID '{job_model.id}' has not reached a terminal state and cannot be archived."
             )

anyscale/job/commands.py

@@ -54,12 +54,12 @@ _SUBMIT_ARG_DOCSTRINGS = {"config": "The config options defining the job."}
     doc_py_example=_SUBMIT_EXAMPLE,
     arg_docstrings=_SUBMIT_ARG_DOCSTRINGS,
 )
-def submit(config: JobConfig, *, _sdk: PrivateJobSDK) -> str:
+def submit(config: JobConfig, *, _private_sdk: Optional[PrivateJobSDK] = None) -> str:
     """Submit a job.
 
     Returns the id of the submitted job.
     """
-    return _sdk.submit(config)
+    return _private_sdk.submit(config)  # type: ignore
 
 
 _STATUS_EXAMPLE = """
@@ -89,12 +89,12 @@ def status(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> JobStatus:
     """Get the status of a job."""
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.status(name=name, job_id=id, cloud=cloud, project=project)
+    return _private_sdk.status(name=name, job_id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _TERMINATE_EXAMPLE = """
@@ -123,7 +123,7 @@ def terminate(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> str:
     """Terminate a job.
@@ -133,7 +133,7 @@ def terminate(
     Returns the id of the terminated job.
     """
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.terminate(name=name, job_id=id, cloud=cloud, project=project)
+    return _private_sdk.terminate(name=name, job_id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _ARCHIVE_EXAMPLE = """
@@ -162,7 +162,7 @@ def archive(
     id: Optional[str] = None,  # noqa: A002
     cloud: Optional[str] = None,
     project: Optional[str] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> str:
     """Archive a job.
@@ -172,7 +172,7 @@ def archive(
     Returns the id of the archived job.
     """
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.archive(name=name, job_id=id, cloud=cloud, project=project)
+    return _private_sdk.archive(name=name, job_id=id, cloud=cloud, project=project)  # type: ignore
 
 
 _WAIT_EXAMPLE = """\
@@ -204,12 +204,12 @@ def wait(
     project: Optional[str] = None,
     state: Union[JobState, str] = JobState.SUCCEEDED,
     timeout_s: float = 1800,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ):
     """"Wait for a job to enter a specific state."""
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    _sdk.wait(
+    _private_sdk.wait(  # type: ignore
         name=name,
         job_id=id,
         cloud=cloud,
@@ -251,12 +251,12 @@ def get_logs(
     run: Optional[str] = None,
     mode: Union[str, JobLogMode] = JobLogMode.TAIL,
     max_lines: Optional[int] = None,
-    _sdk: PrivateJobSDK,
+    _private_sdk: Optional[PrivateJobSDK] = None,
     **_kwargs: Dict[str, Any],
 ) -> str:
     """Query the jobs for a job run."""
     id = _resolve_id_from_args(id, _kwargs)  # noqa: A001
-    return _sdk.get_logs(
+    return _private_sdk.get_logs(  # type: ignore
         job_id=id,
         name=name,
         cloud=cloud,

anyscale/organization_invitation/commands.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Tuple
+from typing import Dict, List, Optional, Tuple
 
 from anyscale._private.sdk import sdk_command
 from anyscale.organization_invitation._private.organization_invitation_sdk import (
@@ -45,13 +45,15 @@ _DELETE_ARG_DOCSTRINGS = {
     arg_docstrings=_CREATE_ARG_DOCSTRINGS,
 )
 def create(
-    emails: List[str], *, _sdk: PrivateOrganizationInvitationSDK
+    emails: List[str],
+    *,
+    _private_sdk: Optional[PrivateOrganizationInvitationSDK] = None
 ) -> Tuple[List[str], List[str]]:
     """Creates organization invitations for the provided emails.
 
     Returns a tuple of successful emails and error messages.
     """
-    return _sdk.create(emails=emails)
+    return _private_sdk.create(emails=emails)  # type: ignore
 
 
 @sdk_command(
@@ -61,13 +63,13 @@ def create(
     arg_docstrings=_LIST_ARG_DOCSTRINGS,
 )
 def list(  # noqa: A001
-    *, _sdk: PrivateOrganizationInvitationSDK
+    *, _private_sdk: Optional[PrivateOrganizationInvitationSDK] = None
 ) -> List[OrganizationInvitation]:
     """Lists organization invitations.
 
     Returns a list of organization invitations.
     """
-    return _sdk.list()
+    return _private_sdk.list()  # type: ignore
 
 
 @sdk_command(
@@ -76,9 +78,11 @@ def list(  # noqa: A001
     doc_py_example=_DELETE_EXAMPLE,
     arg_docstrings=_DELETE_ARG_DOCSTRINGS,
 )
-def delete(email: str, *, _sdk: PrivateOrganizationInvitationSDK) -> str:
+def delete(
+    email: str, *, _private_sdk: Optional[PrivateOrganizationInvitationSDK] = None
+) -> str:
     """Deletes an organization invitation.
 
     Returns the email of the deleted organization invitation.
     """
-    return _sdk.delete(email)
+    return _private_sdk.delete(email)  # type: ignore

anyscale/project/__init__.py

@@ -29,7 +29,7 @@ class ProjectSDK:
     )
     def add_collaborators(  # noqa: F811
         self, cloud: str, project: str, collaborators: List[CreateProjectCollaborator],
-    ) -> str:
+    ) -> None:
         """Batch add collaborators to a project.
         """
-        return self._private_sdk.add_collaborators(cloud, project, collaborators)
+        self._private_sdk.add_collaborators(cloud, project, collaborators)

anyscale/project/_private/project_sdk.py

@@ -11,11 +11,11 @@ from anyscale.project.models import CreateProjectCollaborator
 class PrivateProjectSDK(BaseSDK):
     def add_collaborators(
         self, cloud: str, project: str, collaborators: List[CreateProjectCollaborator]
-    ) -> str:
+    ) -> None:
         cloud_id = self.client.get_cloud_id(cloud_name=cloud, compute_config_id=None)
         project_id = self.client.get_project_id(parent_cloud_id=cloud_id, name=project)
 
-        return self.client.add_project_collaborators(
+        self.client.add_project_collaborators(
             project_id=project_id,
             collaborators=[
                 CreateUserProjectCollaborator(

anyscale/project/commands.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from anyscale._private.sdk import sdk_command
 from anyscale.project._private.project_sdk import PrivateProjectSDK
@@ -49,8 +49,8 @@ def add_collaborators(
     project: str,
     collaborators: List[CreateProjectCollaborator],
     *,
-    _sdk: PrivateProjectSDK
+    _private_sdk: Optional[PrivateProjectSDK] = None
 ) -> str:
     """Batch add collaborators to a project.
     """
-    return _sdk.add_collaborators(cloud, project, collaborators)
+    return _private_sdk.add_collaborators(cloud, project, collaborators)  # type: ignore

anyscale/project_utils.py

@@ -442,7 +442,7 @@ def get_default_project(
             "Please specify a cloud for this command either through the cloud or "
             "compute config arguments."
         )
-    default_project = anyscale_api_client.get_default_project(
+    default_project = anyscale_api_client.get_default_project(  # type: ignore
         parent_cloud_id=parent_cloud_id
     ).result
     return default_project

anyscale/resource_quota/commands.py

@@ -87,11 +87,13 @@ _DISABLE_DOCSTRINGS = {
     arg_docstrings=_CREATE_DOCSTRINGS,
 )
 def create(
-    create_resource_quota: CreateResourceQuota, *, _sdk: PrivateResourceQuotaSDK
+    create_resource_quota: CreateResourceQuota,
+    *,
+    _private_sdk: Optional[PrivateResourceQuotaSDK] = None
 ) -> ResourceQuota:
     """Create a resource quota.
     """
-    return _sdk.create(create_resource_quota)
+    return _private_sdk.create(create_resource_quota)  # type: ignore
 
 
 @sdk_command(
@@ -107,10 +109,10 @@ def list(  # noqa: A001
     is_enabled: Optional[bool] = None,
     max_items: int = 20,
     *,
-    _sdk: PrivateResourceQuotaSDK
+    _private_sdk: Optional[PrivateResourceQuotaSDK] = None
 ) -> List[ResourceQuota]:
     """List resource quotas. """
-    return _sdk.list(name, cloud, creator_id, is_enabled, max_items,)
+    return _private_sdk.list(name, cloud, creator_id, is_enabled, max_items,)  # type: ignore
 
 
 @sdk_command(
@@ -119,10 +121,12 @@ def list(  # noqa: A001
     doc_py_example=_DELETE_EXAMPLE,
     arg_docstrings=_DELETE_DOCSTRINGS,
 )
-def delete(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
+def delete(
+    resource_quota_id: str, *, _private_sdk: Optional[PrivateResourceQuotaSDK] = None
+):
     """Delete a resource quota.
     """
-    return _sdk.delete(resource_quota_id)
+    return _private_sdk.delete(resource_quota_id)  # type: ignore
 
 
 @sdk_command(
@@ -131,10 +135,12 @@ def delete(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
     doc_py_example=_ENABLE_EXAMPLE,
     arg_docstrings=_ENABLE_DOCSTRINGS,
 )
-def enable(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
+def enable(
+    resource_quota_id: str, *, _private_sdk: Optional[PrivateResourceQuotaSDK] = None
+):
     """Enable a resource quota.
     """
-    return _sdk.set_status(resource_quota_id, True)
+    return _private_sdk.set_status(resource_quota_id, True)  # type: ignore
 
 
 @sdk_command(
@@ -143,7 +149,9 @@ def enable(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
     doc_py_example=_DISABLE_EXAMPLE,
     arg_docstrings=_DISABLE_DOCSTRINGS,
 )
-def disable(resource_quota_id: str, *, _sdk: PrivateResourceQuotaSDK):
+def disable(
+    resource_quota_id: str, *, _private_sdk: Optional[PrivateResourceQuotaSDK] = None
+):
     """Disable a resource quota.
     """
-    return _sdk.set_status(resource_quota_id, False)
+    return _private_sdk.set_status(resource_quota_id, False)  # type: ignore