toggle use doc model

Author: esoteric-ephemera

mp_api/client/mprester.py

@@ -1515,7 +1515,11 @@ def get_atom_reference_data(
 
         return {
             dfa: {
-                entry["formula"]: entry["data"][f"{dfa}.energy"].value
+                entry["formula"]: (
+                    entry["data"][f"{dfa}.energy"].value
+                    if self.use_document_model
+                    else entry["data"][dfa]["energy"]["value"]
+                )
                 for entry in _atomic_energies
             }
             for dfa in funcs