-
-
Notifications
You must be signed in to change notification settings - Fork 18.4k
REGR: Don't ignore compiled patterns in replace #35697
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
REGR: Don't ignore compiled patterns in replace #35697
Conversation
This reverts commit 5006980.
Currently failing mypy and not quite sure of the fix. 5006980 did not work.
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@dsaxton looked into that CI error you're getting.
xref python/typing#23
On my machine adding
from typing import Pattern
at the top of pandas/core/internals/managers.py
and then using plain Pattern
in the line you modify fixed the mypy
complaint.
Hope this helps!
Awesome thanks @arw2019 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @dsaxton generally lgtm
@@ -1949,7 +1959,7 @@ def _check_comparison_types( | |||
else: | |||
op = np.vectorize( | |||
lambda x: bool(re.search(b, x)) | |||
if isinstance(x, str) and isinstance(b, str) | |||
if isinstance(x, str) and isinstance(b, (str, Pattern)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
b
is typed as Scalar
, where
Scalar = Union[PythonScalar, PandasScalar]
but b
resolves to Union[builtins.str, builtins.float, Any]
due to unresolved imports and hence mypy is green
but should probably update types and docstring anyway
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @simonjayhawkins. Would the proper type for b be Union[Scalar, Pattern], or should PythonScalar be extended to include Pattern?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I would keep PythonScalar unchanged for now.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Updated. Odd that b is typed as scalar but there's an isinstance check for np.ndarray inside the function, maybe that's not needed.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
the docstring also implies b could be an array. I guess could add ArrayLike while here or leave. either works for me.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'll add it. Actually a can also be a scalar so the type shouldn't only be ArrayLike I don't think:
In [17]: b
Out[17]: array(['a', 'a', 'a'], dtype='<U1')
In [18]: _compare_or_regex_search("a", b)
Out[18]: array([ True, True, True])
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
it looks like the types were originally this way when added, see also #32890 (comment), but changed in #33598
maybe best to keep the changes here minimal for now as this PR is being backported
(and separate PR for ensuring the types are correct and consistent with docstrings.)
I think this is a mypy issue that is fixed in later mypy, but can't see a relevant issue on mypy issue tracker (probably a later typeshed) |
@@ -1928,7 +1941,7 @@ def _compare_or_regex_search( | |||
""" | |||
|
|||
def _check_comparison_types( | |||
result: Union[ArrayLike, bool], a: ArrayLike, b: Scalar, | |||
result: Union[ArrayLike, bool], a: ArrayLike, b: Union[Scalar, Pattern], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
is a Pattern
ever passed through to _check_comparison_types
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I guess it's just for the error message
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @dsaxton lgtm
not a regression, but I guess |
Thanks @dsaxton merging and backporting this as addresses reported regression. followups can be left for 1.2. |
@meeseeksdev backport 1.1.x |
…35765) Co-authored-by: Daniel Saxton <[email protected]>
black pandas
git diff upstream/master -u -- "*.py" | flake8 --diff