Как создать экземпляр defaultdict для выполнения аннотации типа DefaultDict[int, DefaultDict[int, MyClass]]
Я использую Python 3.8 и только что обновил Pyright до 1.1.377, и теперь получаю ошибки ввода для заданий defaultdict(defaultdict).
Код для воспроизведения того, что я имею в виду, будет:
from typing import DefaultDict
from collections import defaultdict
class MyClass:
pass
d: DefaultDict[int, DefaultDict[int, MyClass]] = defaultdict(defaultdict)
d[1][2] = MyClass()
Я получаю ошибку:
error: Expression of type "defaultdict[Unknown, defaultdict[Unknown, defaultdict[Unknown, Unknown]]]" is incompatible with declared type "DefaultDict[int, DefaultDict[int, MyClass]]
Как здесь создать экземпляр defaultdict, чтобы соответствовать объявленному мной типу?
Ответы 2
Если вы посмотрите документацию, это объяснит, почему вы можете предположить, что это должно работать, особенно если вы представляете, что mypy и pyright очень похожи. Пирайт приводит примеры, когда они не обращают внимания на некоторые распространенные проблемы с типами в pyright и mypy в документации ниже.
Ссылка: док
Сказав это,
Проблема, с которой вы столкнулись, связана исключительно с тем, как Pyright обрабатывает вывод типа для defaultdict. Когда вы используете defaultdict(defaultdict) без указания аргументов типа, Pyright, в отличие от mypy, не может вывести правильные типы, что приводит к ошибке, которую вы видите.
Что вы можете сделать, так это использовать старый добрый lambda, чтобы помочь с использованием lambda: lambda func вернет defaultdict правильного типа, который, я надеюсь, вам нужен -> defaultdict[int, MyClass]
вот фрагмент, который поможет вам
from typing import DefaultDict
from collections import defaultdict
class MyClass:
pass
d: DefaultDict[int, DefaultDict[int, MyClass]] = defaultdict(lambda: defaultdict(MyClass))
d[1][2] = MyClass()
Вот простой тест на правильность создания экземпляра defaultdict
d: DefaultDict[int, DefaultDict[int, MyClass]] = defaultdict(lambda: defaultdict(MyClass))
def func(a: DefaultDict[int, DefaultDict[int, MyClass]]):
reveal_type(a) # pyright should now reveal Runtime type is 'defaultdict'
func(d)
Как заметил сопровождающий Pyright по адресу #8839, это ошибка.