From e299d93b25ba93cb585fc7fef12d087ae1ae848a Mon Sep 17 00:00:00 2001 From: MaxMLang Date: Sun, 7 Apr 2024 04:22:38 +0100 Subject: [PATCH] Testing fixes --- .github/workflows/workflow.yml | 3 +++ tests/test_pytector.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/workflow.yml b/.github/workflows/workflow.yml index bda7636..969b3e8 100644 --- 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 index 0ba8074..1814833 100644 --- 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):