fix wrong import

pymc-devs · Oct 1, 2023 · d2ba0fa · d2ba0fa
1 parent 2c01984
commit d2ba0fa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pymc/distributions/discrete.py b/pymc/distributions/discrete.py
@@ -13,7 +13,7 @@
 #   limitations under the License.
 import warnings
 
-from typing import Optional, TypeAlias, Union
+from typing import Optional, Union
 
 import numpy as np
 import numpy.typing as npt
@@ -34,6 +34,7 @@
 )
 from pytensor.tensor.variable import TensorVariable
 from scipy import stats
+from typing_extensions import TypeAlias
 
 import pymc as pm