did the same thing in another place

mosaicml · Jan 18, 2024 · 21aa9e3 · 21aa9e3
1 parent 3ef2b20
commit 21aa9e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/data/finetuning/tasks.py b/llmfoundry/data/finetuning/tasks.py
@@ -62,7 +62,7 @@ def preprocessing_fn(example: Dict) -> Dict[str, str]:
 ChatFormattedDict = Dict[str, List[Dict[str, str]]]
 Conversation = Union[PromptResponseDict, ChatFormattedDict]
 ConversationType = Literal['prompt_response', 'chat']
-TokenizedConversation = Dict[str, List[int | str]]
+TokenizedConversation = Dict[str, List[Union[int, str]]]
 
 
 def _get_conversation_type(conversation_example: Conversation):