fix init

guardrails-ai · Sep 10, 2024 · 86d6ef9 · 86d6ef9
1 parent 7666c85
commit 86d6ef9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/validator/main.py b/validator/main.py
@@ -39,14 +39,13 @@ class LlamaGuard7B(Validator):
     def __init__(
         self,
         policies: Optional[List[str]] = None,
-        validation_method: Optional[str] = "full",
         on_fail: Optional[Callable] = None,
         **kwargs,
     ):
 
         super().__init__(
+            policies=policies,
             on_fail=on_fail,
-            validation_method=validation_method,
             **kwargs,
         )