Skip to content

Commit 4d89ec5

Browse files
committed
Better error when unknown integrator is passed to md
1 parent 948c0ec commit 4d89ec5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

wfl/generate/md/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -306,7 +306,8 @@ def process_step(interval):
306306

307307

308308
def _sample_autopara_wrappable_kwargs(atoms, calculator, steps, dt, **kwargs):
309-
assert kwargs.get("integrator", "Berendsen") in ["Berendsen", "Langevin", "LangevinBAOAB"]
309+
if kwargs.get("integrator", "Berendsen") not in ["Berendsen", "Langevin", "LangevinBAOAB"]:
310+
raise ValueError(f"Unknown integrator {kwargs['integrator']}, not Berendsen, Langevin, or LangevinBAOAB")
310311

311312
calculator = construct_calculator_picklesafe(calculator)
312313

0 commit comments

Comments
 (0)