Running on custom dataset gives me zero division error idx = what_val_to_log % len(val_ray_samplers) ZeroDivisionError: integer division or modulo by zero