From f2d4e1befc77d163c564bc0f3e0db79f954d10a2 Mon Sep 17 00:00:00 2001 From: dqy <1016751306@qq.com> Date: Mon, 22 Apr 2024 17:07:50 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BF=AE=E6=94=B9=E6=A3=80=E6=B5=8B?= =?UTF-8?q?=E7=BB=93=E6=9E=9C=E8=BE=93=E5=87=BA=E8=B7=AF=E5=BE=84=EF=BC=9B?= =?UTF-8?q?=E5=88=A0=E9=99=A4=E6=B5=8B=E8=AF=95=E6=A8=A1=E5=9D=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- detection/backdoor_detection.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/detection/backdoor_detection.py b/detection/backdoor_detection.py index f5139ae..4d539d2 100644 --- a/detection/backdoor_detection.py +++ b/detection/backdoor_detection.py @@ -1,3 +1,5 @@ +# Usage: python backdoor_detection.py + 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 ")