pandas: Incorrect return type for pandas.Timestamp.normalize()

[DannyNemer]

Environment

• Language Server: v2020.12.2
• OS: macOS v10.15.6
• Python: v3.7.9
• Pandas: v1.1.5 (latest)

Expected behavior

Recognize that pandas.Timestamp.normalize() returns an instance of pandas.Timestamp:

Actual behavior

Pylance incorrectly says pandas.Timestamp.normalize() returns None:

Code Snippet

Should not return errors:

import pandas as pd

pd.Timestamp.today().normalize().tz_localize(None)

---

Please let me know how I can improve this bug report because many more will follow (all for typeshed issues). Is this the best repo for such bug reports? (I am following the precedent of Pandas type issues in the release notes.)

[judej]

Thanks for the report @DannyNemer, looks like the type stub needs an update.

[gramster]

It doesn't seem like Timestamp is the issue here, as it says in the stub:

from __future__ import annotations

...
class Timestamp:
    ...
    @classmethod
    def today(cls, tz: Optional[Any] = ...) -> Timestamp: ...
    def normalize(self, *args, **kwargs) -> Timestamp: ...
    def tz_localize(self, tz: Any = ...) -> Timestamp: ...

@erictraut, can you see something I may be missing?

[gramster]

I also see "argument missing for parameter tz" on the line

pd.Timestamp.today()

which makes me wonder if @classmethod is being handled in the stubs.

[jakebailey]

This issue has been fixed in version 2021.5.0, which we've just released. You can find the changelog here: https://github.com/microsoft/pylance-release/blob/main/CHANGELOG.md#202150-5-may-2021