Use forward references for type hints, avoid NameError (#6780)

deepset-ai · Jan 18, 2024 · f47439c · f47439c
1 parent d4f6531
commit f47439c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/haystack/components/generators/chat/hugging_face_local.py b/haystack/components/generators/chat/hugging_face_local.py
@@ -272,7 +272,7 @@ def create_message(
         self,
         text: str,
         index: int,
-        tokenizer: Union[PreTrainedTokenizer, PreTrainedTokenizerFast],
+        tokenizer: Union["PreTrainedTokenizer", "PreTrainedTokenizerFast"],
         prompt: str,
         generation_kwargs: Dict[str, Any],
     ) -> ChatMessage: