From 473d807fddc2c0fd7d9978c1444dfc3ce68b2ec3 Mon Sep 17 00:00:00 2001 From: Zhufeng Pan Date: Wed, 25 Jan 2023 00:18:47 -0800 Subject: [PATCH] Use f-string to replace percent operator to make the logging more robust to data type. (self._num_training/validation_examples are initialized as None) PiperOrigin-RevId: 504482361 --- tensorflow_decision_forests/keras/core.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/tensorflow_decision_forests/keras/core.py b/tensorflow_decision_forests/keras/core.py index 30d0f96..3c51049 100644 --- a/tensorflow_decision_forests/keras/core.py +++ b/tensorflow_decision_forests/keras/core.py @@ -1395,9 +1395,11 @@ def remote_consumes_training_examples_until_eof(iterator): (per_worker_iter,)) if self._verbose >= 1: - tf_logging.info("Training dataset read in %s. Found %d examples.", - datetime.now() - time_begin_train_dataset_reading, - self._num_training_examples) + t = datetime.now() - time_begin_train_dataset_reading + tf_logging.info( + f"Training dataset read in {t}. Found" + f" {self._num_training_examples} examples." + ) # Load the validation dataset if validation_data is not None: @@ -1462,9 +1464,10 @@ def consumes_one_valid_batch(iterator): self.num_validation_examples) if self._verbose >= 1: - tf_logging.info("Validation dataset read in %s. Found %d examples.", - datetime.now() - time_begin_valid_dataset_reading, - self._num_validation_examples) + t = datetime.now() - time_begin_valid_dataset_reading + tf_logging.info( + f"Validation dataset read in {t}. Found" + f" {self._num_validation_examples} examples.") # Train the model # Note: The model should be trained after having collected both the training