fix: fix some error

tests/final_tests_util.py

@@ -1,5 +1,5 @@
 from typing import Tuple
-from git import Repo  
+from git import Repo  # type: ignore
 import random
 from pathlib import Path
 

tests/test_final_tests.py

@@ -1,9 +1,8 @@
 import unittest
-import os
 import shutil
 
 from detection.utils import read_file_content
-from .final_tests_util import *
+from .final_tests_util import clone_repo, Path, inject_random_backdoor
 from detection.Regexdetection import find_dangerous_functions
 from detection.GPTdetection import detectGPT
 
@@ -27,7 +26,11 @@ class TestFinalTests(unittest.TestCase):
         for file in self.all_python_files:
             content = read_file_content(str(file))
             results = find_dangerous_functions(content, ".py")
-            if len(results["high"]) > 0 or len(results["medium"]) > 0 or len(results["low"]) > 0:
+            if (
+                len(results["high"]) > 0
+                or len(results["medium"]) > 0
+                or len(results["low"]) > 0
+            ):
                 detectedNum += 1
                 possibly_dangerous_file.append(file)
         print(detectedNum / self.filesNum)
@@ -39,14 +42,17 @@ class TestFinalTests(unittest.TestCase):
             results = {}
             try:
                 results = detectGPT(content)
-            except Exception as e:
-                print(e)
-            if len(results["high"]) > 0 or len(results["medium"]) > 0 or len(results["low"]) > 0:
+                if (
+                    len(results["high"]) > 0
+                    or len(results["medium"]) > 0
+                    or len(results["low"]) > 0
+                ):
                     GPTdetectedNum += 1
                 print(GPTdetectedNum)
                 self.assertGreaterEqual(GPTdetectedNum, detectedNum)
 
-
+            except Exception as e:
+                print(e)
 
 
 if __name__ == "__main__":