Improve type annotations in naming.py
Improve type annotations in naming.py

PiperOrigin-RevId: 376663398

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