Python 3.9 compatability

bakebrain · Mar 3, 2024 · dab44cd · dab44cd
1 parent f223579
commit dab44cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dspy/signatures/signature.py b/dspy/signatures/signature.py
@@ -188,7 +188,7 @@ class MySignature(Signature):
     pass
 
 
-def ensure_signature(signature: str | Type[Signature]) -> Signature:
+def ensure_signature(signature: Union[str, Type[Signature]]) -> Signature:
     if signature is None:
         return None
     if isinstance(signature, str):

diff --git a/dspy/utils/dummies.py b/dspy/utils/dummies.py
@@ -1,4 +1,5 @@
 import random
+from typing import Union
 from dsp.modules import LM
 import numpy as np
 from dsp.utils.utils import dotdict
@@ -8,7 +9,7 @@
 class DummyLM(LM):
     """Dummy language model for unit testing purposes."""
 
-    def __init__(self, answers: list[str] | dict[str, str], follow_examples: bool = False):
+    def __init__(self, answers: Union[list[str], dict[str, str]], follow_examples: bool = False):
         """Initializes the dummy language model.
 
         Parameters: