python3.9 compatibility

patrick-kidger · Nov 8, 2023 · 5a0f6c8 · 5a0f6c8
1 parent d1bf648
commit 5a0f6c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/diffrax/path.py b/diffrax/path.py
@@ -1,6 +1,6 @@
 import abc
 from dataclasses import field
-from typing import Optional
+from typing import Optional, Union
 
 import equinox as eqx
 import jax
@@ -51,7 +51,7 @@ def evaluate(
         t1: Optional[Scalar] = None,
         left: bool = True,
         use_levy: bool = False,
-    ) -> PyTree | LevyVal:
+    ) -> Union[PyTree, LevyVal]:
         r"""Evaluate the path at any point in the interval $[t_0, t_1]$.
 
         **Arguments:**