add dtype to initialization of y in ConsensusADMM

tttamaki · tttamaki · commit 175a94d5e536 · 2026-03-11T23:30:56.000+09:00
diff --git a/pyproximal/optimization/primal.py b/pyproximal/optimization/primal.py
@@ -2063,7 +2063,7 @@ def ConsensusADMM(  # pylint: disable=invalid-name
     m = len(proxfs)
     x_bar = x0.copy()
     x_bar_old = x0.copy()
-    y = ncp.zeros((m, x0.size))
+    y = ncp.zeros((m, x0.size), dtype=x0.dtype)
 
     # iterate
     for iiter in range(niter):
