I don't know if we have a policy on defining input types, but `uint8` instead of `np.uint8` is probably fine.