-
- Notifications
You must be signed in to change notification settings - Fork 19.3k
add type hint for core/computation/eval #26992
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
Conversation
| I am not fully sure about target variable type and return type. If anyone could take a look and give some feedback, that will be great. Thanks |
WillAyd left a comment
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.
Looks good some comments
pandas/core/computation/eval.py Outdated
| parser: str = 'pandas', | ||
| engine: str = None, | ||
| truediv: bool = True, | ||
| local_dict: dict = None, |
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.
Instead of using dict can you use the generic Dict from the typing module and add subscripts for the key / value 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.
Thanks for this feedback. I looked through the where local_dict be used https://github.com/pandas-dev/pandas/blob/master/pandas/core/computation/scope.py#L103, but I cannot find out what kind of dict is this? [str, str] ? Thanks !
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.
Hmm I'm not intimately familiar with this code but from what I gather should probably be Dict[str, Any] to simulate what can be injected into locals() and/or globals()
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! I will update soon !
pandas/core/computation/eval.py Outdated
| | ||
| | ||
| def _check_expression(expr): | ||
| def _check_expression(expr: Any) -> None: |
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.
Kind of vague from the docstring but looks like the type here should be str (Any should be avoided as it doesn't really offer much in the way of type checking)
pandas/core/computation/eval.py Outdated
| | ||
| | ||
| def _convert_expression(expr): | ||
| def _convert_expression(expr: Any) -> str: |
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 think also string here as well (inferring from actual usage rather than docstring)
pandas/core/computation/eval.py Outdated
| | ||
| | ||
| def _check_for_locals(expr, stack_level, parser): | ||
| def _check_for_locals(expr: Any, stack_level: int, parser: str) -> None: |
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.
Same here
pandas/core/computation/eval.py Outdated
| global_dict: dict = None, | ||
| resolvers: Iterable = (), | ||
| level: int = 0, | ||
| target: Union[np.ndarry, pd.DataFrame, pd.Series] = None, |
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.
Typo on ndarray but this can also return a scalar right?
Codecov Report
@@ Coverage Diff @@ ## master #26992 +/- ## ========================================== - Coverage 91.99% 91.99% -0.01% ========================================== Files 180 180 Lines 50774 50777 +3 ========================================== - Hits 46711 46710 -1 - Misses 4063 4067 +4
Continue to review full report at Codecov.
|
Codecov Report
@@ Coverage Diff @@ ## master #26992 +/- ## ========================================= + Coverage 91.68% 91.99% +0.3% ========================================= Files 182 180 -2 Lines 50246 50779 +533 ========================================= + Hits 46069 46713 +644 + Misses 4177 4066 -111
Continue to review full report at Codecov.
|
WillAyd left a comment
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.
Comments on 3/4 CI failures. Not sure about last one regarding metaclass but may be related to python/mypy#730 if you can take a look
pandas/core/computation/eval.py Outdated
| | ||
| | ||
| def _check_engine(engine): | ||
| def _check_engine(engine: str) -> str: |
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.
engine needs to be Optional[str] since None is valid here (cause of one CI failure)
| | ||
| | ||
| def _check_for_locals(expr, stack_level, parser): | ||
| def _check_for_locals(expr: str, stack_level: int, parser: str) -> None: |
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.
Can just add return None at the end of the body to appease CI
pandas/core/computation/eval.py Outdated
| truediv: bool = True, | ||
| local_dict: Dict[str, Any] = None, | ||
| global_dict: Dict[str, Any] = None, | ||
| resolvers: Iterable = (), |
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 think this needs to be List[Dict] per documentation (may also help CI failure)
| # We want to exclude `inplace=None` as being False. | ||
| if inplace is False: | ||
| return target if target_modified else ret | ||
| else: |
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.
Shouldn't need the else here
| truediv: bool = True, | ||
| local_dict: List[Dict[str, Any]] = None, | ||
| global_dict: List[Dict[str, Any]] = None, | ||
| resolvers: Tuple = (), |
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.
So does this only accept a Tuple of Dict? If so we want to add Tuple[Dict[?, ?], ...] as the annotation (replacing ? with appropriate types) and update the docstring which currently suggests a list is appropriate
| | ||
| | ||
| def _check_resolvers(resolvers): | ||
| def _check_resolvers(resolvers: Iterable) -> None: |
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.
If it's possible we would want to specify the type of object within container classes, so Iterable[<type>] if it can be inferred here
| Removing from 0.25 milestone - @xcz011 if you can merge master and address comments can keep with this though |
Sorry about leaving this open, I think I will try to finish this soon. Thanks |
| Can you merge master? Couldn't see old build failures |
| closing as stale, if you'd like to continue pls ping to reopen |
git diff upstream/master -u -- "*.py" | flake8 --diff