Fix linting complaints.

guardrails-ai · Oct 31, 2024 · 05e89b6 · 05e89b6
1 parent 43a917c
commit 05e89b6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/validator/main.py b/validator/main.py
@@ -46,7 +46,7 @@ def __init__(
         # There are some spurious loading complaints with TFDistilBert models.
         # See https://discuss.huggingface.co/t/message-some-layers-from-the-model-were-not-used/1972/7
 
-    def validate(self, value: Any, metadata: Dict = None) -> ValidationResult:
+    def validate(self, value: Any, metadata: Optional[Dict] = None) -> ValidationResult:
         """Validates that the text is free from biases related to age, gender, sex, ethnicity, religion, etc."""
         if isinstance(value, str):
             value = [value,]  # Ensure we're always passing lists of strings into the classifier.

diff --git a/validator/post-install.py b/validator/post-install.py
@@ -1,4 +1,4 @@
-print("post-install starting...")
 from transformers import pipeline
+print("post-install starting...")
 _ = pipeline("text-classification", "d4data/bias-detection-model")
 print("post-install complete!")