Skip to content

Commit 9b8ab75

Browse files
committed
fixup asv
1 parent 6ba030c commit 9b8ab75

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

asv_bench/benchmarks/groupby.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -552,26 +552,26 @@ class groupby_period(object):
552552
# GH 14338
553553
goal_time = 0.2
554554

555+
def make_grouper(self, N):
556+
return pd.period_range('1900-01-01', freq='D', periods=N)
557+
555558
def setup(self):
556559
N = 10000
557-
self.grouper = self.make_grouper()
560+
self.grouper = self.make_grouper(N)
558561
self.df = pd.DataFrame(np.random.randn(N, 2))
559562

560-
def make_grouper(self):
561-
return pd.period_range('1900-01-01', freq='D', periods=N)
562-
563563
def time_groupby_sum(self):
564564
self.df.groupby(self.grouper).sum()
565565

566566

567567
class groupby_datetime(groupby_period):
568-
def make_grouper(self):
568+
def make_grouper(self, N):
569569
return pd.date_range('1900-01-01', freq='D', periods=N)
570570

571571

572572
class groupby_datetimetz(groupby_period):
573-
def make_grouper(self):
574-
return pd.date_range('1900-01-01', freq='D', periods=N
573+
def make_grouper(self, N):
574+
return pd.date_range('1900-01-01', freq='D', periods=N,
575575
tz='US/Central')
576576

577577
#----------------------------------------------------------------------

0 commit comments

Comments
 (0)