Использование mypy с методом pandas `to_dict` ⇐ Python

[Anonymous]

Вот упрощенная версия моего кода:

Код: Выделить всё

import pandas as pd
from typing import Dict

df = pd.DataFrame(
{"year": [2024, 2025], "my_output": [1, 2], "foo": ["a", "b"]}
)

df.set_index("year", inplace=True)

row_dict = df.to_dict("index")

ouput_dict: Dict[int, int] = {
key: val["my_output"] for key, val in row_dict.items()
}

Когда я запускаю mypy, я получаю следующую ошибку:
Ключевое выражение в словарном понимании имеет несовместимый тип «Hashable»; ожидаемый тип "int"
Как я могу сообщить mypy, что ключи, которые я установил с помощью столбца "год", являются целыми числами?

Подробнее здесь: https://stackoverflow.com/questions/784 ... ict-method