Use numpy 'atleast_1d' in place of homebuilt function.