fix inverse_phat_square

lehner · Dec 11, 2023 · 3f122d6 · 3f122d6
1 parent d44bfb4
commit 3f122d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/gpt/algorithms/optimize/fourier_accelerate.py b/lib/gpt/algorithms/optimize/fourier_accelerate.py
@@ -44,7 +44,7 @@ def __init__(self, grid, base, dimensions=None):
             self.weight += c_mu_l
 
         # special consideration for zero
-        self.weight[coor[0:1]] = (2.0 * np.pi) ** 2.0 / np.prod(
+        self.weight[[[0] * grid.nd]] = (2.0 * np.pi) ** 2.0 / np.prod(
             [grid.gdimensions[mu] for mu in dimensions]
         ) ** (2.0 / len(dimensions))