change exit() to sys.exit() (#248)

m3dev · Sep 11, 2021 · a8e66c1 · a8e66c1
1 parent 49f23d8
commit a8e66c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/gokart/run.py b/gokart/run.py
@@ -38,7 +38,7 @@ def _try_tree_info(cmdline_args):
     else:
         raise ValueError(f'--tree-info-mode must be "simple" or "all", but "{mode}" is passed.')
     logger.info(f'output tree info: {output_path}')
-    exit()
+    sys.exit()
 
 
 def _try_to_delete_unnecessary_output_file(cmdline_args: List[str]):
@@ -49,7 +49,7 @@ def _try_to_delete_unnecessary_output_file(cmdline_args: List[str]):
                 logger.info('delete-unnecessary-output-files is not support s3/gcs.')
             else:
                 gokart.delete_local_unnecessary_outputs(task)
-            exit()
+            sys.exit()
 
 
 def _try_get_slack_api(cmdline_args: List[str]) -> Optional[gokart.slack.SlackAPI]:

diff --git a/gokart/testing/check_if_run_with_empty_data_frame.py b/gokart/testing/check_if_run_with_empty_data_frame.py
@@ -1,4 +1,5 @@
 import logging
+import sys
 from typing import List, Optional
 
 import luigi
@@ -70,7 +71,7 @@ def _test_run_with_empty_data_frame(cmdline_args: List[str], test_run_params: te
 
     test_logger.info('gokart test results:\n' + '\n'.join(s.format() for s in test_status_list))
     if any(s.fail() for s in test_status_list):
-        exit(1)
+        sys.exit(1)
 
 
 def try_to_run_test_for_empty_data_frame(cmdline_args: List[str]):
@@ -80,4 +81,4 @@ def try_to_run_test_for_empty_data_frame(cmdline_args: List[str]):
     if test_run_params.pandas:
         cmdline_args = [a for a in cmdline_args if not a.startswith('--test-run-')]
         _test_run_with_empty_data_frame(cmdline_args=cmdline_args, test_run_params=test_run_params)
-        exit(0)
+        sys.exit(0)