update tests

Author: aaravnavani

guardrails/validator_service.py

@@ -3,7 +3,7 @@
 import os
 from concurrent.futures import ProcessPoolExecutor
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 from guardrails.classes.history import Iteration
 from guardrails.datatypes import FieldValidation
@@ -60,7 +60,7 @@ def perform_correction(
         results: List[FailResult],
         value: Any,
         validator: Validator,
-        on_fail_descriptor: OnFailAction,
+        on_fail_descriptor: Union[OnFailAction, str],
     ):
         if on_fail_descriptor == OnFailAction.FIX:
             # FIXME: Should we still return fix_value if it is None?

guardrails/validators/reading_time.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional
+from typing import Any, Callable, Dict, Optional
 
 from guardrails.logger import logger
 from guardrails.validator_base import (
@@ -28,7 +28,7 @@ class ReadingTime(Validator):
         reading_time: The maximum reading time in minutes.
     """
 
-    def __init__(self, reading_time: int, on_fail: Optional[str] = None):
+    def __init__(self, reading_time: int, on_fail: Optional[Callable] = None):
         super().__init__(
             on_fail=on_fail,
             reading_time=reading_time,