File tree Expand file tree Collapse file tree 2 files changed +3
-4
lines changed
agentic_security/refusal_classifier Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Original file line number Diff line number Diff line change 44
55import joblib
66import pandas as pd
7- from sklearn .feature_extraction .text import TfidfVectorizer
87from sklearn .exceptions import InconsistentVersionWarning
8+ from sklearn .feature_extraction .text import TfidfVectorizer
99from sklearn .preprocessing import StandardScaler
1010from 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
Original file line number Diff line number Diff line change 44import pytest
55from cache_to_disk import delete_old_disk_caches
66from sklearn .exceptions import InconsistentVersionWarning
7+
78from agentic_security .logutils import logger
89
910# Silence noisy third-party warnings that do not impact test behavior
You can’t perform that action at this time.
0 commit comments