Skip to content

Commit

Permalink
Update run.py
Browse files Browse the repository at this point in the history
remove unused sampling_context
  • Loading branch information
gustavor101 authored Aug 7, 2024
1 parent 0b3c20e commit dc4639c
Showing 1 changed file with 4 additions and 9 deletions.
13 changes: 4 additions & 9 deletions examples/lammps/spectral_abf/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
# %%
import argparse
import sys

import numpy
from lammps import lammps

Expand All @@ -18,6 +19,7 @@
from pysages.colvars import DihedralAngle
from pysages.methods import HistogramLogger, SpectralABF


# %%
def generate_context(args="", script="adp.lmp", store_freq=1):
"""
Expand All @@ -39,19 +41,15 @@ def get_args(argv):
("kokkos", "k", bool, True, "Whether to use Kokkos acceleration"),
("log-steps", "l", int, 2e3, "Number of simulation steps for logging"),
]
parser = argparse.ArgumentParser(
description="Example script to run pysages with lammps"
)
parser = argparse.ArgumentParser(description="Example script to run pysages with lammps")

for name, short, T, val, doc in available_args:
if T is bool:
action = "store_" + str(val).lower()
parser.add_argument("--" + name, "-" + short, action=action, help=doc)
else:
convert = (lambda x: int(float(x))) if T is int else T
parser.add_argument(
"--" + name, "-" + short, type=convert, default=T(val), help=doc
)
parser.add_argument("--" + name, "-" + short, type=convert, default=T(val), help=doc)

return parser.parse_args(argv)

Expand All @@ -77,9 +75,6 @@ def main(argv):
)
method = SpectralABF(cvs, grid)
callback = HistogramLogger(args.log_steps)
sampling_context = SamplingContext(
method, generate_context, callback=callback, context_args=context_args
)
raw_result = pysages.run(
method,
generate_context,
Expand Down

0 comments on commit dc4639c

Please sign in to comment.