detection/backdoor_detection.py

@@ -1,3 +1,5 @@
+# Usage: python backdoor_detection.py <code_path> <output_format>
+
 import os
 import re
 import sys
@@ -70,7 +72,7 @@ def output_results(
     results: Dict[str, List[Tuple[int, str]]], output_format: str, file_path: str
 ):
     # Create the 'results' directory if it does not exist
-    results_dir = "../results"
+    results_dir = "../results/code"
     if not os.path.exists(results_dir):
         os.makedirs(results_dir)
 
@@ -146,10 +148,6 @@ def process_path(path: str, output_format: str):
         sys.exit(1)
 
 
-def test():
-    print("hello world")
-
-
 def main():
     if len(sys.argv) < 3:
         print("Usage: python backdoor_detection.py <path> <output_format>")