Stupid error, forgot to import numpy in pytrol