Use callable in lazy imports to avoid type errors
Use callable in lazy imports to avoid type errors

PiperOrigin-RevId: 412268214

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