TypeError: 'axis' must be None, an integer or a tuple of integers

[segalinc]

getting TypeError: 'axis' must be None, an integer or a tuple of integers when running normalized function on an single image using same example simple_inference code