fix layer is not define on toutpt_collective tests
pep8