Mypy does not recognize torch.cuda.amp

I could not post all the code but basically the mypy does not pass because it does not recognize the amp.autocast() function and throw an error

error: Module has no attribute "autocast"

from torch.cuda import amp 
 with amp.autocast():
     # some code

here is my code to run mypy :

python -m mypy --ignore-missing-imports . || exit 1

Is it possible to deactivate this error