Skip to content

Commit 3fd1b2c

Browse files
graceful exit for aqua cli
1 parent 621ec25 commit 3fd1b2c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

ads/aqua/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,4 +41,3 @@ def set_log_level(log_level: str):
4141
logger.error(
4242
f"Aqua is not available for this notebook session {NB_SESSION_OCID}."
4343
)
44-
exit()

ads/aqua/cli.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
66
import os
77

8-
from ads.aqua import ENV_VAR_LOG_LEVEL, set_log_level
8+
from ads.aqua import ENV_VAR_LOG_LEVEL, set_log_level, ODSC_MODEL_COMPARTMENT_OCID
99
from ads.aqua.deployment import AquaDeploymentApp
1010
from ads.aqua.evaluation import AquaEvaluationApp
1111
from ads.aqua.finetune import AquaFineTuningApp
@@ -41,3 +41,6 @@ def __init__(
4141
'WARNING', 'ERROR', and 'CRITICAL'.
4242
"""
4343
set_log_level(log_level)
44+
# gracefully exit if env var is not set
45+
if not ODSC_MODEL_COMPARTMENT_OCID:
46+
exit()

0 commit comments

Comments
 (0)