Got around the global samples error by using the input/output features.
Removed test.py