Skip to content

Commit 79feb99

Browse files
TomAugspurgerjreback
authored andcommitted
BUG: Fixup bench for deprecation (#27207)
1 parent 3e4f000 commit 79feb99

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

asv_bench/benchmarks/categoricals.py

+2-9
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
import warnings
2-
31
import numpy as np
42
import pandas as pd
53
import pandas.util.testing as tm
@@ -133,15 +131,10 @@ def setup(self):
133131

134132
self.s_str = pd.Series(tm.makeCategoricalIndex(N, ncats)).astype(str)
135133
self.s_str_cat = self.s_str.astype('category')
136-
with warnings.catch_warnings(record=True):
137-
self.s_str_cat_ordered = self.s_str.astype('category',
138-
ordered=True)
139-
134+
self.s_str_cat_ordered = self.s_str_cat.cat.as_ordered()
140135
self.s_int = pd.Series(np.random.randint(0, ncats, size=N))
141136
self.s_int_cat = self.s_int.astype('category')
142-
with warnings.catch_warnings(record=True):
143-
self.s_int_cat_ordered = self.s_int.astype('category',
144-
ordered=True)
137+
self.s_int_cat_ordered = self.s_int_cat.cat.as_ordered()
145138

146139
def time_rank_string(self):
147140
self.s_str.rank()

0 commit comments

Comments
 (0)