Skip to content

Commit f19664f

Browse files
committed
fix(pc):
1 parent b3ae002 commit f19664f

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

agentic_security/refusal_classifier/model.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44

55
import joblib
66
import pandas as pd
7-
from sklearn.feature_extraction.text import TfidfVectorizer
87
from sklearn.exceptions import InconsistentVersionWarning
8+
from sklearn.feature_extraction.text import TfidfVectorizer
99
from sklearn.preprocessing import StandardScaler
1010
from sklearn.svm import OneClassSVM
1111

@@ -91,9 +91,7 @@ def load_model(self):
9191
self.model = joblib.load(f)
9292

9393
# Load vectorizer
94-
with pkg_resources.open_binary(
95-
package, "tfidf_vectorizer.joblib"
96-
) as f:
94+
with pkg_resources.open_binary(package, "tfidf_vectorizer.joblib") as f:
9795
self.vectorizer = joblib.load(f)
9896

9997
# Load scaler

tests/conftest.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import pytest
55
from cache_to_disk import delete_old_disk_caches
66
from sklearn.exceptions import InconsistentVersionWarning
7+
78
from agentic_security.logutils import logger
89

910
# Silence noisy third-party warnings that do not impact test behavior

0 commit comments

Comments
 (0)