fix

Samsung · Jan 20, 2024 · 741ea04 · 741ea04
1 parent 1fa1a30
commit 741ea04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/experiment/src/prepare_data.py b/experiment/src/prepare_data.py
@@ -6,9 +6,9 @@
 def execute_scanner(dataset_location: str, result_location_str, j, use_ml=False):
     """Execute CredSweeper as a separate process to make sure no global states is shared with training script"""
     dir_path = os.path.dirname(os.path.realpath(__file__)) + "/.."
-    command = f"{sys.executable} -m credsweeper --path {dataset_location}/data --save-json {result_location_str} -j {j} --severity critical"
+    command = f"{sys.executable} -m credsweeper --path {dataset_location}/data --save-json {result_location_str} -j {j}"
     if not use_ml:
-        command += "--no-filters --ml_threshold 0"
+        command += " --ml_threshold 0"
     subprocess.call(command, shell=True, cwd=dir_path, stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT)