used older union operator for legacy support

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