diff --git a/detection/backdoor_detection.py b/detection/backdoor_detection.py index 4d539d2..457ca93 100644 --- a/detection/backdoor_detection.py +++ b/detection/backdoor_detection.py @@ -72,7 +72,8 @@ 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/code" + # 这里如果集成测试的话应该设置为./ + results_dir = "./results/code" if not os.path.exists(results_dir): os.makedirs(results_dir) diff --git a/detection/requirements_detection.py b/detection/requirements_detection.py index 2daba0d..1350aa0 100644 --- a/detection/requirements_detection.py +++ b/detection/requirements_detection.py @@ -72,8 +72,9 @@ def check_vulnerabilities(requirements, vulnerabilities, output_format): results.append(f"OK: {req_name}=={req_version} is not affected.") else: results.append(f"OK: {req_name} not found in the vulnerability database.") + # 集成测试这里应该修改为./ output_results( - "../results/requirements/results." + output_format, results, output_format + "./results/requirements/results." + output_format, results, output_format )