change | to Union (py 39)

enoch3712 · Nov 20, 2024 · ac5233a · ac5233a
1 parent 11d056b
commit ac5233a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extract_thinker/process.py b/extract_thinker/process.py
@@ -47,7 +47,7 @@ def __init__(self):
         self.masking_strategy: Optional[AbstractMaskingStrategy] = None
         self.llm: Optional[LLM] = None
 
-    def add_masking_llm(self, model: Optional[str | LLM] = None, strategy: Optional[MaskContract] = MaskingStrategy.SIMPLE_PLACEHOLDER):
+    def add_masking_llm(self, model: Optional[Union[str, LLM]] = None, strategy: Optional[MaskContract] = MaskingStrategy.SIMPLE_PLACEHOLDER):
         if isinstance(model, LLM):
             self.llm = model
         elif model is not None: