Skip to content

Commit ec936fc

Browse files
add option to approx neb common maker to specify different endpoint relax maker
1 parent 8d8d3d8 commit ec936fc

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/atomate2/common/flows/approx_neb.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,12 @@ class CommonApproxNebMaker(Maker):
3535
Name of the workflow
3636
host_relax_maker : Maker
3737
Optional, a maker to relax the input host structure.
38-
image_relax_maker : maker
38+
image_relax_maker : Maker
3939
Required, a maker to relax the ApproxNEB endpoints and images.
40+
endpoint_relax_maker : Maker or None (default)
41+
Optional maker to relax the endpoints that could differ from the
42+
relax maker used on the intermediate images.
43+
If None, this is set to `image_relax_maker`.
4044
selective_dynamics_scheme : "fix_two_atoms" (default) or None
4145
If "fix_two_atoms", uses the default selective dynamics scheme of ApproxNEB,
4246
wherein the migrating ion and the ion farthest from it are the only
@@ -51,6 +55,7 @@ class CommonApproxNebMaker(Maker):
5155
name: str = "ApproxNEB"
5256
host_relax_maker: Maker | None = None
5357
image_relax_maker: Maker = None
58+
endpoint_relax_maker: Maker | None = None
5459
selective_dynamics_scheme: Literal["fix_two_atoms"] | None = "fix_two_atoms"
5560
min_hop_distance: float | bool = True
5661

@@ -114,7 +119,7 @@ def make(
114119
working_ion=working_ion,
115120
endpoint_coords=inserted_coords_dict,
116121
inserted_coords_combo=inserted_coords_combo,
117-
relax_maker=self.image_relax_maker,
122+
relax_maker=self.endpoint_relax_maker or self.image_relax_maker,
118123
)
119124

120125
# run pathfinder (and selective dynamics) to get image structure input

0 commit comments

Comments
 (0)