fix: 修改检测模块引入

This commit is contained in:
dqy 2024-05-29 21:24:08 +08:00
parent 99b481059b
commit 41b16c53bc

View File

@ -1,7 +1,7 @@
import unittest
import warnings
from detection.backdoor_detection import find_dangerous_functions
from detection.__main__ import find_dangerous_functions
from detection.GPTdetection import detectGPT
import os
@ -83,5 +83,6 @@ class TestBackdoorDetection(unittest.TestCase):
self.assertEqual(len(results["medium"]), 0)
self.assertEqual(len(results["low"]), 0)
if __name__ == "__main__":
unittest.main()