diff --git a/validator/main.py b/validator/main.py index 4b57e33..8cf1f87 100644 --- 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, )