Allow --module_imports as comma separated values.
Allow --module_imports as comma separated values.

PiperOrigin-RevId: 310362179

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets