check_py_code is only disabled when running the test test_dropout.py