* Fix scope closure

Joshuaalbert · Sep 6, 2024 · 7167051 · 7167051
1 parent 1cb92f3
commit 7167051
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsa2000_cal/dsa2000_cal/common/jvp_linear_op.py b/dsa2000_cal/dsa2000_cal/common/jvp_linear_op.py
@@ -132,7 +132,7 @@ def _get_results_type(primal_out: jax.Array):
 
             def _adjoint_promote_dtypes(co_tangent: jax.Array, dtype: jnp.dtype):
                 if co_tangent.dtype != dtype:
-                    warnings.warn(f"Promoting co-tangent dtype from {co_tangent.dtype} to {primal_out.dtype}.")
+                    warnings.warn(f"Promoting co-tangent dtype from {co_tangent.dtype} to {dtype}.")
                 return co_tangent.astype(dtype)
 
             # v @ J