use sys exit instead of builtins

Author: VipulMascarenhas

ads/aqua/cli.py

@@ -4,6 +4,7 @@
 # Copyright (c) 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 import os
+import sys
 
 from ads.aqua import ENV_VAR_LOG_LEVEL, set_log_level, ODSC_MODEL_COMPARTMENT_OCID
 from ads.aqua.deployment import AquaDeploymentApp
@@ -43,4 +44,4 @@ def __init__(
         set_log_level(log_level)
         # gracefully exit if env var is not set
         if not ODSC_MODEL_COMPARTMENT_OCID:
-            exit()
+            sys.exit(0)

tests/unitary/with_extras/aqua/test_cli.py

@@ -56,9 +56,10 @@ def test_aquacommand(self, name, arg, expected, mock_setting_log):
             AquaCommand()
         mock_setting_log.assert_called_with(expected)
 
-    def test_aqua_command_without_compartment_env_var(self):
+    @patch("sys.exit")
+    def test_aqua_command_without_compartment_env_var(self, mock_exit):
         os.environ.pop("ODSC_MODEL_COMPARTMENT_OCID", None)
         reload(ads.aqua)
         reload(ads.aqua.cli)
-        with pytest.raises(SystemExit):
-            AquaCommand()
+        AquaCommand()
+        mock_exit.assert_called_with(0)