xiaohei/model_garden
1
0
Fork 0
mirror of https://github.com/GoogleCloudPlatform/vertex-ai-samples.git synced 2026-09-26 14:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a no-op message to check_quota if it fails.

Browse Source 
MG_DOCKER_CODES_PIPER_ORIGIN_REV_ID: 792654121
This commit is contained in:
Rayan Dasoriya
2025-08-08 09:32:08 -07:00
committed by Copybara-Service
parent 58dab0b1bb
commit 44390cbd99
2 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/community/model_garden/docker_source_codes/model_oss/notebook_util/common_util.py
+12 -3
View File
@@ -494,9 +494,12 @@ def get_quota(project_id: str, region: str, resource_id: str) -> int:
f" --service={service_endpoint} --consumer=projects/{project_id}"
f" --filter='{service_endpoint}/{resource_id}' --format=json"
)
process = subprocess.run(
command, shell=True, capture_output=True, text=True, check=True
)
try:
process = subprocess.run(
command, shell=True, capture_output=True, text=True, check=True
)
except subprocess.CalledProcessError:
return -2
if process.returncode == 0:
quota_data = json.loads(process.stdout)
else:
@@ -662,6 +665,12 @@ def check_quota(
" to check quota in a region or request additional quota for "
"your project."
)
if quota == -2:
print(
"This is a no-op for now. Please check the quota manually to verify if"
" you have enough quota for your use case."
)
return
if quota == -1:
raise ValueError(
f"Quota not found for: {resource_id} in {region}."
notebooks/community/model_garden/docker_source_codes/notebook_util/common_util.py
+12 -3
View File
@@ -494,9 +494,12 @@ def get_quota(project_id: str, region: str, resource_id: str) -> int:
f" --service={service_endpoint} --consumer=projects/{project_id}"
f" --filter='{service_endpoint}/{resource_id}' --format=json"
)
process = subprocess.run(
command, shell=True, capture_output=True, text=True, check=True
)
try:
process = subprocess.run(
command, shell=True, capture_output=True, text=True, check=True
)
except subprocess.CalledProcessError:
return -2
if process.returncode == 0:
quota_data = json.loads(process.stdout)
else:
@@ -662,6 +665,12 @@ def check_quota(
" to check quota in a region or request additional quota for "
"your project."
)
if quota == -2:
print(
"This is a no-op for now. Please check the quota manually to verify if"
" you have enough quota for your use case."
)
return
if quota == -1:
raise ValueError(
f"Quota not found for: {resource_id} in {region}."