Skip to content
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

in get_type_hints(Appliction), TypeError: <class 'traitlets.traitlets.Dict'> is not a generic class #904

Open
spchamp opened this issue Apr 25, 2024 · 1 comment

Comments

@spchamp
Copy link

spchamp commented Apr 25, 2024

The expression get_type_hints(Application) raises an exception, given the class traitlets.config.application.Application or a subclass.

Minimal Example

file traitlets_debug_00.py

from traitlets.config.application import Application
from typing_extensions import get_type_hints

if __name__ == "__main__":
    hints = get_type_hints(Application)

Traceback from __main__:

Traceback (most recent call last):
  File ".../traitlets_debug_00.py", line 6, in <module>
    hints = get_type_hints(Application)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/lib64/python3.11/typing.py", line 2305, in get_type_hints
    value = _eval_type(value, base_globals, base_locals)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/lib64/python3.11/typing.py", line 359, in _eval_type
    return t._evaluate(globalns, localns, recursive_guard)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/lib64/python3.11/typing.py", line 857, in _evaluate
    eval(self.__forward_code__, globalns, localns),
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "<string>", line 1, in <module>
  File "/usr/lib64/python3.11/typing.py", line 344, in inner
    return func(*args, **kwds)
           ^^^^^^^^^^^^^^^^^^^
  File "/usr/lib64/python3.11/typing.py", line 1807, in __class_getitem__
    _check_generic(cls, params, len(cls.__parameters__))
  File "/usr/lib64/python3.11/typing.py", line 271, in _check_generic
    raise TypeError(f"{cls} is not a generic class")
TypeError: <class 'traitlets.traitlets.Dict'> is not a generic class

Analysis

in traitlets.traitlets line 3842, the Dict TraitType is defined like as follows:

class Dict(Instance["dict[K, V]"]):
   ...

In traitlets.config.application, line 403, the type is applied as:

    subcommands: dict[str, t.Any] | Dict[str, t.Any] = Dict()

The Dict TraitType has not been defined as a subscriptable type, however.

Patch?

At an instance scope, the value of the subcommands attribute should generally be a Python dict, while at a class scope, the attribute would be represented with a TraitType descriptor, namely an instance of the Dict TraitType.

As per the note about the syntax of the subcommands value in the source code where the attr is defined, the subcommands attr could be defined as follows:

    subcommands: dict[str, tuple[Callable, str]] = Dict()

In effect, this would remove the class-scoped Dict TraitType descriptor from the type hint for the instance-scoped subcommands attribute value, also providing some more detail about the syntax of the attribute value.

The initialization of the subcommands descriptor object would not be changed with this definition.

Workarounds?

As one approach, the following can be used to patch the annotation in the base Application class, while providing an Application-like class with the patched type hint:

from traitlets.config.application import Application
from traitlets.traitlets import Dict
from collections.abc import Callable
import typing as t

class ApplicationShim(Application):
    subcommands: dict[str, tuple[t.Union[str, Callable[[t.Self], t.Self]], str]] = Dict()

Application.__annotations__['subcommands'] = ApplicationShim.__annotations__['subcommands']

The type hint in this patch is referenced to how the attribute value is applied in Application.initialize_subcommand() with examples for how the subcommands value is defined as a class variable, in the following:

  • IPython.core.profileapp.ProfileApp
  • jupyter_client.kernelspecapp.KernelSpecApp
  • ipykernel.kernelapp.IPKernelApp
  • ipyparallel.cluster.app.IPCluster

To apply the patched type hint insofar as may be visible within editor environments for instance, the class ApplicationShim could be used as a base class in lieu of Application. The patch onto the base subcommands annotation would be the main thing, however, insofar as for application with get_type_hints()

Impacts?

This exception is encountered when calling get_type_hints(Application).

This may affect some reflection-oriented user code and could possibly affect type analysis for the Application class

@spchamp
Copy link
Author

spchamp commented Apr 25, 2024

of course, the Dict component of the original type hint would be applicable for when the subcommands attribute is accessed at the class scope, as a descriptor, e.g Application.subcommands

Not related to this issue, just to ask: I don't suppose anyone's got an idea for drafting a PEP about how type hint interpretation may differ for a descriptor type accessed at the class scope, as the descriptor object itself -- typically returned from __get__ in this case -- or accessed at the instance scope, vis a vis the return value from __get__ when the instance arg is not None?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant