From f19664f95c40134e2679c69cc4c2e8ee51ed66a6 Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Sun, 30 Nov 2025 18:32:58 +0200 Subject: [PATCH] fix(pc): --- agentic_security/refusal_classifier/model.py | 6 ++---- tests/conftest.py | 1 + 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/agentic_security/refusal_classifier/model.py b/agentic_security/refusal_classifier/model.py index abca117..cd42e75 100644 --- a/agentic_security/refusal_classifier/model.py +++ b/agentic_security/refusal_classifier/model.py @@ -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 diff --git a/tests/conftest.py b/tests/conftest.py index 2db7641..c42ed95 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -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