Как справиться с ошибкой pyright: "Оптимизатор" не является известным членом модуля?

Question

Моя цель - дать подсказку типа для экземпляра любого класса, который наследуется от torch.optim.Optimizer, как показано в следующем примере.

import torch.optim as optim

def custom_function(optimizer: optim.Optimizer) -> None:
    pass

Я использую pyright как средство проверки статического типа, которое выдает ошибку: "Оптимизатор" не является известным членом модуля.

Есть предложения, как справиться или исправить это?