fix single atom energies in script, close #326

mir-group · Oct 24, 2022 · 1adb753 · 1adb753
1 parent f6e457f
commit 1adb753
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flare/scripts/otf_train.py b/flare/scripts/otf_train.py
@@ -285,6 +285,8 @@ def get_sgp_calc(flare_config):
             sae_dct
         ), "'single_atom_energies' should be the same length as 'species'"
         single_atom_energies = {i: sae_dct[i] for i in range(n_species)}
+    else:
+        single_atom_energies = {i: 0 for i in range(n_species)}
 
     sgp = SGP_Wrapper(
         kernels=kernels,