[pspec] correct keys

2021-04-28 19:21:09 +02:00
parent 200091d01b
commit 7d6faba644
1 changed files with 1 additions and 1 deletions
@@ -838,7 +838,7 @@ def main(arg):
        # 2D power spectra -------------------------------------------------------------
        ns = 2 ** clvl
        f = "_%%(i)0%dd" % (np.floor(np.log10(ns)) + 1)
-        for v in var_names:
+        for v in list(pcubes2.keys()):
            for i in range(ns):
                pspec, kbins, pspec2, fbins = pspectrum(
                    pcubes2[v][:, :, i], cubes_k["k"][:, :, i], kbins, knorm, arg.fbins