Allow test to pass when there are unrelated logging.warning() calls
Allow test to pass when there are unrelated logging.warning() calls

PiperOrigin-RevId: 379362851

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