From 86d6ef9879bd8de9841c3f7d3c156c5e7a406ab0 Mon Sep 17 00:00:00 2001
From: David Tam <david@guardrailsai.com>
Date: Tue, 10 Sep 2024 10:51:26 -0700
Subject: [PATCH] fix init

---
 validator/main.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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,
         )