[Data] Make sure num_gpus provide to Ray Data is appropriately passed to ray.remote call

[Data] Make sure num_gpus provide to Ray Data is appropriately passed to ray.remote call #5885