Testing fixes

MaxMLang · Apr 7, 2024 · e299d93 · e299d93
1 parent 197a13c
commit e299d93
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/workflow.yml b/.github/workflows/workflow.yml
@@ -20,6 +20,9 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install setuptools wheel twine
+    - name: Install dependencies
+      run: |
+        pip install validators
     - name: Run tests
       run: |
         python -m unittest discover -s tests

diff --git a/tests/test_pytector.py b/tests/test_pytector.py
@@ -1,6 +1,6 @@
 import unittest
-from pytector import PromptInjectionDetector
-
+import validators
+import pytector
 class TestPromptInjectionDetector(unittest.TestCase):
 
     def test_initialization_with_predefined_model(self):