Update scripts/crystal/eval_crystalgflownet.py

alexhernandezgarcia · Jan 25, 2024 · b28c5a4 · b28c5a4
1 parent 10ace79
commit b28c5a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/crystal/eval_crystalgflownet.py b/scripts/crystal/eval_crystalgflownet.py
@@ -195,7 +195,7 @@ def main(args):
                 min_angle=0.0,
                 max_angle=1.0,
             )
-            energies = env.oracle(env.statebatch2proxy(x_sampled))
+            energies = env.oracle(env.states2proxy(x_sampled))
             dct = {"energy": energies.tolist()}
             pickle.dump(dct, open(tmp_dir / f"randomcrystals_samples_{i}.pkl", "wb"))