also prefix with $

mlverse · Dec 11, 2023 · c8a23d7 · c8a23d7
1 parent 3b24703
commit c8a23d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/callbacks.R b/R/callbacks.R
@@ -435,7 +435,7 @@ luz_callback_lr_scheduler <- luz_callback(
       lr_scheduler(optimizer, ...)
     }
     self[[call_on]] <- function() {
-      if ("metrics" %in% names(formals(scheduler$step))) {
+      if ("metrics" %in% names(formals(self$scheduler$step))) {
         self$scheduler$step(current_loss)
       } else {
         self$scheduler$step()