pep8, test work with device=cpu