Введите подсказки для подмножества класса ⇐ Python

[Anonymous]

Я пишу функции, которые действуют на pandas.Series с помощью DatetimeIndex. Я могу делать подсказки по типу, например:

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

import pandas as pd

def filter_year(s: pd.Series, year: int) -> pd.Series:
keep = s.index.year == year
return s[keep]

Это работает нормально, но редактор жалуется, что невозможно получить доступ к атрибуту «год» для класса «Индекс». Причина в том, что редактор ожидает любой индекс.
Есть ли способ указать, что s имеет DatetimeIndex (который является подклассом Index)?

Подробнее здесь: https://stackoverflow.com/questions/787 ... t-of-class