mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-07-10 05:28:37 +02:00
fix: issue #156 - move test file to tests dir
This commit is contained in:
@@ -17,3 +17,4 @@ inv/
|
|||||||
scripts/
|
scripts/
|
||||||
docx/
|
docx/
|
||||||
agentic_security.toml
|
agentic_security.toml
|
||||||
|
/venv
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
|
|
||||||
from agentic_security.refusal_classifier import RefusalClassifier
|
from agentic_security.refusal_classifier.model import RefusalClassifier
|
||||||
|
|
||||||
classifier = RefusalClassifier()
|
classifier = RefusalClassifier()
|
||||||
classifier.load_model()
|
classifier.load_model()
|
||||||
|
|||||||
+1
-1
@@ -3,7 +3,7 @@ from unittest.mock import patch
|
|||||||
import pandas as pd
|
import pandas as pd
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from .model import RefusalClassifier
|
from agentic_security.refusal_classifier.model import RefusalClassifier
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
Reference in New Issue
Block a user