mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-24 06:09:55 +02:00
fix(pc):
This commit is contained in:
@@ -4,8 +4,8 @@ import warnings
|
||||
|
||||
import joblib
|
||||
import pandas as pd
|
||||
from sklearn.feature_extraction.text import TfidfVectorizer
|
||||
from sklearn.exceptions import InconsistentVersionWarning
|
||||
from sklearn.feature_extraction.text import TfidfVectorizer
|
||||
from sklearn.preprocessing import StandardScaler
|
||||
from sklearn.svm import OneClassSVM
|
||||
|
||||
@@ -91,9 +91,7 @@ class RefusalClassifier:
|
||||
self.model = joblib.load(f)
|
||||
|
||||
# Load vectorizer
|
||||
with pkg_resources.open_binary(
|
||||
package, "tfidf_vectorizer.joblib"
|
||||
) as f:
|
||||
with pkg_resources.open_binary(package, "tfidf_vectorizer.joblib") as f:
|
||||
self.vectorizer = joblib.load(f)
|
||||
|
||||
# Load scaler
|
||||
|
||||
@@ -4,6 +4,7 @@ import warnings
|
||||
import pytest
|
||||
from cache_to_disk import delete_old_disk_caches
|
||||
from sklearn.exceptions import InconsistentVersionWarning
|
||||
|
||||
from agentic_security.logutils import logger
|
||||
|
||||
# Silence noisy third-party warnings that do not impact test behavior
|
||||
|
||||
Reference in New Issue
Block a user