Fix small bug in activity grad

thebuckleylab · Jun 24, 2024 · 79d24ea · 79d24ea
1 parent 1d35387
commit 79d24ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jpc/_core/_grads.py b/jpc/_core/_grads.py
@@ -39,7 +39,7 @@ def _neg_activity_grad(
 
     """
     generator, output, input = args
-    dFdzs = grad(pc_energy_fn, argnums=1)(
+    dFdzs = grad(energy_fn, argnums=1)(
         generator,
         activities,
         output,