fix affect

Author: juliasloan25

src/shared_utilities/utils.jl

@@ -543,7 +543,7 @@ end
 
 """
     NaNCheckCallback(nancheck_frequency::Union{AbstractFloat, Dates.Period},
-                        output_dir, start_date, t_start; model, dt)
+                        start_date, t_start, dt)
 
 Constructs a DiscreteCallback which counts the number of NaNs in the state
 and produces a warning if any are found.
@@ -556,12 +556,9 @@ and produces a warning if any are found.
 - `dt`: The timestep of the model (optional), used to check for consistency.
 
 The callback uses `ClimaDiagnostics.EveryCalendarDtSchedule` to determine when
-to save checkpoints based on the `nancheck_frequency`. The schedule is
-initialized with the `start_date` and `t_start` to ensure that the first
-checkpoint is saved at the correct time.
-
-The `save_checkpoint` function is called with the current state vector `u`, the
-current time `t`, and the `output_dir` to save the checkpoint to disk.
+to check for NaNs based on the `nancheck_frequency`. The schedule is
+initialized with the `start_date` and `t_start` to ensure that it is first
+called at the correct time.
 """
 function NaNCheckCallback(
     nancheck_frequency::Union{AbstractFloat, Dates.Period},
@@ -597,9 +594,7 @@ function NaNCheckCallback(
     cond = let schedule = schedule
         (u, t, integrator) -> schedule(integrator)
     end
-    affect! = let output_dir = output_dir, model = model
-        (integrator) -> count_nans_state(integrator.u)
-    end
+    affect! = (integrator) -> count_nans_state(integrator.u)
 
     SciMLBase.DiscreteCallback(cond, affect!)
 end