File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -2318,7 +2318,7 @@ def _transform_fast(self, func):
23182318 func = getattr (self ,func )
23192319 values = func ().values
23202320 counts = self .count ().values
2321- values = np .repeat (values , counts )
2321+ values = np .repeat (values , com . _ensure_platform_int ( counts ) )
23222322
23232323 # the values/counts are repeated according to the group index
23242324 indices = self .indices
Original file line number Diff line number Diff line change @@ -803,7 +803,8 @@ def test_transform_fast(self):
803803
804804 grp = df .groupby ('id' )['val' ]
805805
806- expected = pd .Series (np .repeat (grp .mean ().values , grp .count ().values ),index = df .index )
806+ values = np .repeat (grp .mean ().values , com ._ensure_platform_int (grp .count ().values ))
807+ expected = pd .Series (values ,index = df .index )
807808 result = grp .transform (np .mean )
808809 assert_series_equal (result ,expected )
809810
You can’t perform that action at this time.
0 commit comments