diff --git a/agentic_security/refusal_classifier/hybrid_classifier.py b/agentic_security/refusal_classifier/hybrid_classifier.py index 38075e6..9021f4c 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: