def nanmean(a, axis): # don't call nan_to_num in here, unless you check that # occlusion_test.py still works after you do it! result = nanmean_impl(a, axis=axis) return result