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
python -m mypy --ignore-missing-imports . || exit 1
Is it possible to deactivate this error