Fix bug with loading module.

Adibvafa · Sep 12, 2024 · 32557c2 · 32557c2
1 parent b014942
commit 32557c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CodonTransformer/CodonPrediction.py b/CodonTransformer/CodonPrediction.py
@@ -278,7 +278,7 @@ def load_tokenizer(tokenizer_path: Optional[str] = None) -> PreTrainedTokenizerF
 
 def tokenize(
     batch: List[Dict[str, Any]],
-    tokenizer: Optional[PreTrainedTokenizerFast, str] = None,
+    tokenizer: Union[PreTrainedTokenizerFast, str] = None,
     max_len: int = 2048,
 ) -> BatchEncoding:
     """