Skip to content

Commit 4f14b24

Browse files
fix benchmark failure with numpy 1.20+ (#39795)
1 parent 6835c85 commit 4f14b24

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

asv_bench/benchmarks/series_methods.py

+6
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
import numpy as np
44

5+
from pandas.compat.numpy import np_version_under1p20
6+
57
from pandas import Categorical, NaT, Series, date_range
68

79
from .pandas_vb_common import tm
@@ -143,6 +145,10 @@ class IsInLongSeriesLookUpDominates:
143145

144146
def setup(self, dtype, MaxNumber, series_type):
145147
N = 10 ** 7
148+
149+
if not np_version_under1p20 and dtype in ("Int64", "Float64"):
150+
raise NotImplementedError
151+
146152
if series_type == "random_hits":
147153
np.random.seed(42)
148154
array = np.random.randint(0, MaxNumber, N)

0 commit comments

Comments
 (0)