Skip to content

BUG: groupby.transform/agg caching *args with numba engine #41656

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
May 26, 2021

Conversation

mroeschke
Copy link
Member

@mroeschke mroeschke added Bug numba numba-accelerated operations labels May 25, 2021
@mroeschke mroeschke added this to the 1.3 milestone May 25, 2021
@@ -1131,10 +1131,16 @@ def _transform_with_numba(self, data, func, *args, engine_kwargs=None, **kwargs)
group_keys = self.grouper._get_group_keys()

numba_transform_func = numba_.generate_numba_transform_func(
tuple(args), kwargs, func, engine_kwargs
kwargs, func, engine_kwargs
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

shouldn't knwargs be handled in the same way?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

numba doesn't support **kwargs in udf's currently. It's passed here to raise an error if kwargs is not empty

@jreback jreback merged commit 532166e into pandas-dev:master May 26, 2021
@jreback
Copy link
Contributor

jreback commented May 26, 2021

thanks @mroeschke

@mroeschke mroeschke deleted the bug/numba_groupby_args branch May 26, 2021 03:26
TLouf pushed a commit to TLouf/pandas that referenced this pull request Jun 1, 2021
JulianWgs pushed a commit to JulianWgs/pandas that referenced this pull request Jul 3, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Bug numba numba-accelerated operations
Projects
None yet
Development

Successfully merging this pull request may close these issues.

BUG: numba function being cached with arguments
2 participants