sangge/BackDoorBuster
sangge/BackDoorBuster
1
0
Fork 0
Code Issues Pull Requests Projects 1 Activity

fix: 修改检测模块引入

Browse Source
This commit is contained in:
dqy 2024-05-29 21:24:08 +08:00
parent 99b481059b
commit 41b16c53bc
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_backdoor_detection.py
View File

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