We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 52a9341 commit 50e12c1Copy full SHA for 50e12c1
src/fes_ml/alchemical/modifications/ml_base_modification.py
@@ -57,7 +57,7 @@ def create_cv(
57
# Add ML forces to the CV
58
ml_forces = []
59
for force_id, force in enumerate(system.getForces()):
60
- if force.getName() == "TorchForce":
+ if force.getName() == "TorchForce" or force.getName() == "PythonForce":
61
ml_forces.append((force_id, force))
62
63
ml_vars = []
0 commit comments