fix(warnings):

This commit is contained in:
Alexander Myasoedov
2025-11-30 18:30:55 +02:00
parent 8ddfec303f
commit b3ae0026fb
3 changed files with 43 additions and 19 deletions
+26 -18
View File
@@ -1,9 +1,11 @@
import importlib.resources as pkg_resources
import os
import warnings
import joblib
import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.exceptions import InconsistentVersionWarning
from sklearn.preprocessing import StandardScaler
from sklearn.svm import OneClassSVM
@@ -70,27 +72,33 @@ class RefusalClassifier:
"""
Load the trained model, vectorizer, and scaler from disk.
"""
try:
self.model = joblib.load(self.model_path)
self.vectorizer = joblib.load(self.vectorizer_path)
self.scaler = joblib.load(self.scaler_path)
except FileNotFoundError:
# Load from package resources
package = (
__package__ # This should be 'agentic_security.refusal_classifier'
)
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=InconsistentVersionWarning)
try:
self.model = joblib.load(self.model_path)
self.vectorizer = joblib.load(self.vectorizer_path)
self.scaler = joblib.load(self.scaler_path)
except FileNotFoundError:
# Load from package resources
package = (
__package__ # This should be 'agentic_security.refusal_classifier'
)
# Load model
with pkg_resources.open_binary(package, "oneclass_svm_model.joblib") as f:
self.model = joblib.load(f)
# Load model
with pkg_resources.open_binary(
package, "oneclass_svm_model.joblib"
) as f:
self.model = joblib.load(f)
# Load vectorizer
with pkg_resources.open_binary(package, "tfidf_vectorizer.joblib") as f:
self.vectorizer = joblib.load(f)
# Load vectorizer
with pkg_resources.open_binary(
package, "tfidf_vectorizer.joblib"
) as f:
self.vectorizer = joblib.load(f)
# Load scaler
with pkg_resources.open_binary(package, "scaler.joblib") as f:
self.scaler = joblib.load(f)
# Load scaler
with pkg_resources.open_binary(package, "scaler.joblib") as f:
self.scaler = joblib.load(f)
def is_refusal(self, text):
"""