ArgonneCPAC
diff --git a/‎diffhtwo/experimental/n_mag_opt.py‎
Lines changed: 1 addition & 1 deletion b/‎diffhtwo/experimental/n_mag_opt.py‎
Lines changed: 1 addition & 1 deletion
@@ -743,7 +743,7 @@ def _opt_update(opt_state, i):
     0,
     0,
     None,
-    None,
+    0,
 )
 _loss_kern_multi_z = jjit(
     vmap(
Original file line number	Diff line number	Diff line change
`@@ -743,7 +743,7 @@ def _opt_update(opt_state, i):`
`743`	`743`	`0,`
`744`	`744`	`0,`
`745`	`745`	`None,`
`746`		`- None,`
	`746`	`+ 0,`
`747`	`747`	`)`
`748`	`748`	`_loss_kern_multi_z = jjit(`
`749`	`749`	`vmap(`