ENH: Export (a subset of?) pandas._typing for type checking

[caneff]

Feature Type

• Adding new functionality to pandas

• Changing existing functionality in pandas

• Removing existing functionality in pandas

Problem Description

There are public functions and methods whose arguments are types that are not actually exported. This makes it hard to propogate those types to other functions that then call the pandas ones.

For instance, merge has a how argument that has a type _typing.MergeHow = Literal['cross', 'inner', 'left', 'outer', 'right'], but since _typing is protected, there is no good way to take it as an argument and instead I have to say

def foo(df: pd.DataFrame, ...., how: str):
  ...
  assert how in ['cross', 'inner', 'left', 'outer', 'right']
  pd.merge(..., how=how)

For my type checker to be OK with it. This is both annoyingly verbose and fragile to updates of Pandas

Feature Description

Add a typing module that exposes a (possible subset) of _typing.

I say possible subset because from looking at the _typing module there are clearly types that are internal usage only and I'm guessing we don't want to have them public so that they can be changed easier.

I would propose the subset be all types that are used as arguments of public functions and methods.

This way my function above could have;

import pandas as pd
import pandas.typing as pd_typing
def foo(df: pd.DataFrame, ..., how: pd_typing.MergeHow):
   ...
   pd.merge(...., how=how)

and have everything work.

Alternative Solutions

Technically these types are "available" when imported by other modules, so you can access MergeHow via pandas.core.reshape.merge.MergeHow or pandas.core.frame.MergeHow but those are just imports from _typing imported to be used by those modules themselves, not something users should rely on.

Other alternatives

A) Split the public ones out of _typing into typing, could from typing import * in _typing if we don't want to rewrite everywhere the newly public types are used.

B) Just make all of typing public. As someone who is not heavy into Pandas internals I have no strong opinion here but my guess is that there are internal types that we don't want public.

Additional Context

I'm more than willing to take this PR myself I just want feedback about whether this would be accepted.

[twoertwein]

A subset of _typing exposed in #48577

[caneff]

Ok so any blocker to me exposing some of the literals in that same place then? Like MergeHow and CorrelationMethod are the two we have run into the most at Google when trying to upgrade to pandas 2 but I'm sure I can find others looking through that file that make sense in the same way.

…

On Fri, Sep 22, 2023, 8:10 AM Torsten Wörtwein ***@***.***> wrote: A subset of _typing exposed in #48577 <#48577> — Reply to this email directly, view it on GitHub <#55231 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABG4SH6X34LA4TPXXBBUE5TX3V53HANCNFSM6AAAAAA5B3KDFE> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[twoertwein]

I think pandas.api.typing currently contains mostly classes (they are definitely stable) but I'm honestly not sure how stable the type aliases in pandas._typing are (both their name and value).

We also have pandas-stubs (but you can obviously not import from there).

@Dr-Irv @rhshadrach

[Dr-Irv]

Not sure why you can't just use from pandas._typing import MergeHow to address your current issue.

Alternatively, we could have a pandas/typing.py for the ones we are willing to export (or maybe put all of those in pandas.api.typing) and keep pandas/_typing.py for internal usage.

There is also the issue of creating the documentation for things like MergeHow. If we put them in a public place (whether pandas.typing or pandas.api.typing, then they would need to be documented, so that changing the names of the types (if that should ever happen) would be known to the public.

So, this isn't just an issue of whether we start exposing these literals, it's also a documentation issue.

[caneff]

`pandas._typing` isn't a solution because it is protected and not expected to be imported from other packages. It is never a good idea to break that agreement, especially for functionality this mundane (type checking of inputs to already public methods). If we are exposing types through arguments of public methods, we should be able to access those types in a way that isn't through whatever `pandas.core` file happened to import them, or by `typing.get_type_hints` or the like. I'm fine adding documentation mentioned as well if that is the only objection.

…

On Fri, Sep 22, 2023, 9:53 AM Irv Lustig ***@***.***> wrote: Not sure why you can't just use from pandas._typing import MergeHow to address your current issue. Alternatively, we could have a pandas/typing.py for the ones we are willing to export (or maybe put all of those in pandas.api.typing) and keep pandas/_typing.py for internal usage. There is also the issue of creating the documentation for things like MergeHow. If we put them in a public place (whether pandas.typing or pandas.api.typing, then they would need to be documented, so that changing the names of the types (if that should ever happen) would be known to the public. So, this isn't just an issue of whether we start exposing these literals, it's also a documentation issue. — Reply to this email directly, view it on GitHub <#55231 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABG4SHY3UMFXHWIKKILIMS3X3WJ4NANCNFSM6AAAAAA5B3KDFE> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[rhshadrach]

I think pandas.api.typing would make sense for this, and find having pandas.typing alongside pandas.api.typing a bit confusing.

But the main concern is with evolution of type hints alongside the library. How do we go about changing a name or removing a type-hint without breaking user code? If type-hints were strings, then I don't think we'd be at risk. For example:

from __future__ import annotations


def foo(x) -> bar:
    return x + 1

foo(1)

is valid Python. But they currently aren't by default and without the future import this code fails. I don't believe we have any way to signal to a user that a type-hint is deprecated or going to change.

[caneff]

I agree with that sentiment but presumably it is equivalent even if we don't export these. If you are going to widen the literals to more values, then some code that had actually copied the literal definition of or someone importing from _typing is the same. If you are reducing it to fewer values, then they will break the same way and can be deprecated by detecting usage of the value basically. Not exposing them saves a deprecation only on the case of a rename, to the expense of no one else being able to access it without violating the protectedness of _typing. I agree with your point but it is a different one from my original question

…

On Fri, Sep 22, 2023, 8:28 PM Richard Shadrach ***@***.***> wrote: I think pandas.api.typing would make sense for this, and find having pandas.typing alongside pandas.api.typing a bit confusing. But the main concern is with evolution of type hints alongside the library. How do we go about changing a name or removing a type-hint without breaking user code? If type-hints were strings, then I don't think we'd be at risk. For example: from __future__ import annotations def foo(x) -> bar: return x + 1 foo(1) is valid Python. But they currently aren't by default and without the future import this code fails. I don't believe we have any way to signal to a user that a type-hint is deprecated or going to change. — Reply to this email directly, view it on GitHub <#55231 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABG4SH2IREQIVCYWAXFQQW3X3YULXANCNFSM6AAAAAA5B3KDFE> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[twoertwein]

Two thoughts:

• We could separate the stable pandas.api.typing and create a new pandas.api.typing.aliases that is documented to be public but NOT stable (names might change without warnings between releases and the values too) - typing pandas is already slow, I would like to avoid more friction (adding deprecation/future warnings for typing changes)
• Most people probably do not use the panda's internal annotations (only bare pyright people or people who specifically add a py.typed), pylance and presumably most mypy people use pandas-stubs: if we expose type aliases, we need to ensure that they are in sync between pandas and pandas-stubs!

[caneff]

1. That's acceptable to me. 2. At Google we haven't adopted stubs yet (but it so on my list). We depend on the bare types ATM and given our size of codebase it will be a slow transition. Do pandas-stubs not just import from typing anyway? Isn't that how we keep things in sync?

…

On Sat, Sep 23, 2023, 10:56 AM Torsten Wörtwein ***@***.***> wrote: Two thoughts: - We could separate the stable panas.api.typing and create a new pandas.api.typing.aliases that is documented to be public but NOT stable (names might change without warnings between releases and the values too) - typing pandas is already slow, I would like to avoid more friction (adding deprecation/future warnings for typing changes) - Most people probably do not use the panda's internal annotations (only bare pyright people or people who specifically add a py.typed), pylance and presumably most mypy people use pandas-stubs: if we expose type aliases, we need to ensure that they are in sync between pandas and pandas-stubs! — Reply to this email directly, view it on GitHub <#55231 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABG4SHYNLHFP7ISSUUSK47DX332BJANCNFSM6AAAAAA5B3KDFE> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[Dr-Irv]

2. At Google we haven't adopted stubs yet (but it so on my list). We depend on the bare types ATM and given our size of codebase it will be a slow transition.

I think you will find that using pandas-stubs will help improve your code, because the typing checks a lot more things than what is available with just using the library's typing, which is really meant for checking the internal code consistency.

Do pandas-stubs not just import from typing anyway? Isn't that how we keep things in sync?

See https://github.com/pandas-dev/pandas-stubs#differences-between-type-declarations-in-pandas-and-pandas-stubs

[caneff]

Yes I agree. But with the scale of our codebase makes it daunting to say the least. Any bit of new typing results in a lot of new typing violations to be handled. Any suggestions for how to maybe introduce them piecewise? Also how would the stubs help with my current issue though? Do the stubs intentionally export MergeHow anywhere?

[Dr-Irv]

Yes I agree. But with the scale of our codebase makes it daunting to say the least. Any bit of new typing results in a lot of new typing violations to be handled. Any suggestions for how to maybe introduce them piecewise?

With mypy and pyright, I am pretty sure that you can specify the files to include/exclude for type checking.

Also how would the stubs help with my current issue though? Do the stubs intentionally export MergeHow anywhere?

I think you could do something like this:

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from pandas._typing import MergeHow

def foo(df: pd.DataFrame, ..., how: "MergeHow"):
   ...
   pd.merge(...., how=how)