Skip to content

Commit

Permalink
Fixed 'add_methods' when the 'select' argument is specified. (pandas-…
Browse files Browse the repository at this point in the history
  • Loading branch information
HonzaOsobne authored and gfyoung committed Jul 22, 2017
1 parent d884e51 commit 28622c5
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pandas/core/ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,13 +149,15 @@ def names(x):
def add_methods(cls, new_methods, force, select, exclude):
if select and exclude:
raise TypeError("May only pass either select or exclude")
methods = new_methods

if select:
select = set(select)
methods = {}
for key, method in new_methods.items():
if key in select:
methods[key] = method
new_methods = methods

if exclude:
for k in exclude:
new_methods.pop(k, None)
Expand Down

0 comments on commit 28622c5

Please sign in to comment.