tweak ETA to rate

mlverse · Aug 31, 2023 · f7e160f · f7e160f
1 parent 4812a4a
commit f7e160f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -69,4 +69,4 @@ Collate:
     'reexports.R'
     'serialization.R'
 Remotes:
-  mlverse/torch
+    mlverse/torch
diff --git a/R/callbacks.R b/R/callbacks.R
@@ -238,7 +238,7 @@ luz_callback_progress <- luz_callback(
     # Specially for testing purposes we don't want to have the
     # progress bar showing the ETA.
     if (getOption("luz.show_progress_bar_eta", TRUE)) {
-      if (!is.na(format)) {
+      if (!is.na(total)) {
         format <- paste0(format, " - ETA: :eta")
       } else {
         format <- paste0(format, " - Rate: :tick_rate iter/s")