Use typing.Union for support with older Python versions

shivashankarv · Nov 14, 2023 · 703de02 · 703de02
1 parent c883bc5
commit 703de02
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lit_gpt/tokenizer.py b/lit_gpt/tokenizer.py
@@ -1,12 +1,12 @@
 import json
 from pathlib import Path
-from typing import Optional
+from typing import Optional, Union
 
 import torch
 
 
 class Tokenizer:
-    def __init__(self, checkpoint_dir: Path | str) -> None:
+    def __init__(self, checkpoint_dir: Union[Path, str]) -> None:
         checkpoint_dir = Path(checkpoint_dir)
         if not checkpoint_dir.exists():
             raise NotADirectoryError(f"The checkpoint directory does not exist: {str(checkpoint_dir)}")