Skip to content

Commit d20c1a3

Browse files
authored
Merge pull request #159 from dogakantarci/move-test-model
fix: issue #156 - move test_model.py file to /tests directory
2 parents 21180b5 + ebac62e commit d20c1a3

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,4 @@ inv/
1717
scripts/
1818
docx/
1919
agentic_security.toml
20+
/venv

agentic_security/probe_actor/refusal.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from abc import ABC, abstractmethod
22

3-
from agentic_security.refusal_classifier import RefusalClassifier
3+
from agentic_security.refusal_classifier.model import RefusalClassifier
44

55
classifier = RefusalClassifier()
66
classifier.load_model()

agentic_security/refusal_classifier/test_model.py renamed to tests/refusal_classifier/test_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import pandas as pd
44
import pytest
55

6-
from .model import RefusalClassifier
6+
from agentic_security.refusal_classifier.model import RefusalClassifier
77

88

99
@pytest.fixture

0 commit comments

Comments
 (0)