Skip to content

Commit 5d4c011

Browse files
update test data to reflect emmet schema changes
1 parent 97014ed commit 5d4c011

File tree

3 files changed

+10
-0
lines changed

3 files changed

+10
-0
lines changed

src/atomate2/common/jobs/approx_neb.py

+6
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,9 @@ def collate_results(
160160

161161
hop_dict[combo_name] = NebResult(
162162
images=[calc["structure"] for calc in hop if calc["structure"] is not None],
163+
image_indices=[
164+
idx for idx, calc in enumerate(hop) if calc["structure"] is not None
165+
],
163166
initial_images=[
164167
calc["initial_structure"]
165168
for calc in hop
@@ -548,6 +551,9 @@ def collate_images_single_hop(
548551

549552
return NebResult(
550553
images=[calc["structure"] for calc in calcs if calc["structure"] is not None],
554+
image_indices=[
555+
idx for idx, calc in enumerate(calcs) if calc["structure"] is not None
556+
],
551557
initial_images=[
552558
calc["initial_structure"]
553559
for calc in calcs
Binary file not shown.

tests/vasp/flows/test_approx_neb.py

+4
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,10 @@ def test_approx_neb_flow(mock_vasp, clean_dir, vasp_test_dir):
113113
for idx, energy in enumerate(ref_hop.energies)
114114
)
115115

116+
from monty.serialization import dumpfn
117+
118+
dumpfn(output["collate_results"], "/Users/aaronkaplan/Desktop/temp_aneb.json.gz")
119+
116120
assert all(
117121
getattr(output["collate_results"], f"{direction}_barriers")[k]
118122
== pytest.approx(getattr(ref_results, f"{direction}_barriers")[k])

0 commit comments

Comments
 (0)