Skip to content

Commit b3ae002

Browse files
committed
fix(warnings):
1 parent 8ddfec3 commit b3ae002

File tree

3 files changed

+46
-22
lines changed

3 files changed

+46
-22
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,3 +19,4 @@ docx/
1919
agentic_security.toml
2020
/venv
2121
*.csv
22+
agentic_security/agents/operator_agno.py

agentic_security/refusal_classifier/model.py

Lines changed: 29 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
import importlib.resources as pkg_resources
22
import os
3+
import warnings
34

45
import joblib
56
import pandas as pd
67
from sklearn.feature_extraction.text import TfidfVectorizer
8+
from sklearn.exceptions import InconsistentVersionWarning
79
from sklearn.preprocessing import StandardScaler
810
from sklearn.svm import OneClassSVM
911

@@ -70,27 +72,33 @@ def load_model(self):
7072
"""
7173
Load the trained model, vectorizer, and scaler from disk.
7274
"""
73-
try:
74-
self.model = joblib.load(self.model_path)
75-
self.vectorizer = joblib.load(self.vectorizer_path)
76-
self.scaler = joblib.load(self.scaler_path)
77-
except FileNotFoundError:
78-
# Load from package resources
79-
package = (
80-
__package__ # This should be 'agentic_security.refusal_classifier'
81-
)
82-
83-
# Load model
84-
with pkg_resources.open_binary(package, "oneclass_svm_model.joblib") as f:
85-
self.model = joblib.load(f)
86-
87-
# Load vectorizer
88-
with pkg_resources.open_binary(package, "tfidf_vectorizer.joblib") as f:
89-
self.vectorizer = joblib.load(f)
90-
91-
# Load scaler
92-
with pkg_resources.open_binary(package, "scaler.joblib") as f:
93-
self.scaler = joblib.load(f)
75+
with warnings.catch_warnings():
76+
warnings.filterwarnings("ignore", category=InconsistentVersionWarning)
77+
try:
78+
self.model = joblib.load(self.model_path)
79+
self.vectorizer = joblib.load(self.vectorizer_path)
80+
self.scaler = joblib.load(self.scaler_path)
81+
except FileNotFoundError:
82+
# Load from package resources
83+
package = (
84+
__package__ # This should be 'agentic_security.refusal_classifier'
85+
)
86+
87+
# Load model
88+
with pkg_resources.open_binary(
89+
package, "oneclass_svm_model.joblib"
90+
) as f:
91+
self.model = joblib.load(f)
92+
93+
# Load vectorizer
94+
with pkg_resources.open_binary(
95+
package, "tfidf_vectorizer.joblib"
96+
) as f:
97+
self.vectorizer = joblib.load(f)
98+
99+
# Load scaler
100+
with pkg_resources.open_binary(package, "scaler.joblib") as f:
101+
self.scaler = joblib.load(f)
94102

95103
def is_refusal(self, text):
96104
"""

tests/conftest.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,25 @@
11
import os
2+
import warnings
23

34
import pytest
45
from cache_to_disk import delete_old_disk_caches
5-
6+
from sklearn.exceptions import InconsistentVersionWarning
67
from agentic_security.logutils import logger
78

9+
# Silence noisy third-party warnings that do not impact test behavior
10+
warnings.filterwarnings("ignore", category=InconsistentVersionWarning)
11+
try:
12+
from langchain_core._api import LangChainDeprecationWarning
13+
14+
warnings.filterwarnings("ignore", category=LangChainDeprecationWarning)
15+
except Exception: # pragma: no cover - fallback for older langchain versions
16+
warnings.filterwarnings(
17+
"ignore",
18+
category=DeprecationWarning,
19+
module=r"langchain\\.agents",
20+
message=r".*langchain_core.pydantic_v1.*",
21+
)
22+
823

924
def pytest_runtest_setup(item):
1025
if "slow" in item.keywords and not os.getenv("RUN_SLOW_TESTS"):

0 commit comments

Comments
 (0)