Merge pull request #18054 from pedrobrs:use_io_utils_for_print_or_logging

PiperOrigin-RevId: 536751484

Author: tensorflower-gardener

keras/utils/dataset_utils.py

@@ -670,11 +670,13 @@ def get_training_or_validation_split(samples, labels, validation_split, subset):
 
     num_val_samples = int(validation_split * len(samples))
     if subset == "training":
-        print(f"Using {len(samples) - num_val_samples} files for training.")
+        io_utils.print_msg(
+            f"Using {len(samples) - num_val_samples} " f"files for training."
+        )
         samples = samples[:-num_val_samples]
         labels = labels[:-num_val_samples]
     elif subset == "validation":
-        print(f"Using {num_val_samples} files for validation.")
+        io_utils.print_msg(f"Using {num_val_samples} files for validation.")
         samples = samples[-num_val_samples:]
         labels = labels[-num_val_samples:]
     else: