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