Update utils.py

Update `UFuncTypeError` to be compatible with `numpy>=2.0`.
theislab · Jun 18, 2024 · e3b6f4c · e3b6f4c
1 parent f21651c
commit e3b6f4c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/scvelo/plotting/utils.py b/scvelo/plotting/utils.py
@@ -5,7 +5,6 @@
 
 import numpy as np
 import pandas as pd
-from numpy.core._exceptions import UFuncTypeError
 from pandas import Index
 from scipy import stats
 from scipy.sparse import issparse
@@ -24,6 +23,12 @@
 from scvelo.tools.utils import strings_to_categoricals
 from . import palettes
 
+try:
+    from numpy.core._exceptions import UFuncTypeError
+except ModuleNotFoundError:
+    from numpy._core._exceptions import UFuncTypeError
+
+
 """helper functions"""