diff --git a/agentic_security/refusal_classifier/hybrid_classifier.py b/agentic_security/refusal_classifier/hybrid_classifier.py index 9021f4c..38075e6 100644 --- a/agentic_security/refusal_classifier/hybrid_classifier.py +++ b/agentic_security/refusal_classifier/hybrid_classifier.py @@ -86,7 +86,7 @@ class HybridRefusalClassifier: detector: RefusalDetector, weight: float = 1.0, name: str | None = None, - ) -> "HybridRefusalClassifier": + ) -> HybridRefusalClassifier: """Add a detection method with specified weight. Args: