https://github.com/tskit-dev/msprime
Tip revision: 5860d93b7c183e09fe93d31185e1fb7230ad6d60 authored by mergify[bot] on 12 November 2021, 18:08:17 UTC
Merge pull request #1912 from jeromekelleher/v1.0.3-final-updates
Merge pull request #1912 from jeromekelleher/v1.0.3-final-updates
Tip revision: 5860d93
run_old_msprime.py
"""
Runs the parameters for a given simulation in an input
pickle file and writes the output tree sequence to
an output file.
"""
import pickle
import sys
import msprime
if __name__ == "__main__":
assert msprime.__version__ == "0.7.4"
if len(sys.argv) != 4:
raise ValueError(
"Usage: python3 run_old_msprime.py <num_replicates> "
"<params file> <output prefix>"
)
num_replicates = int(sys.argv[1])
with open(sys.argv[2], "rb") as f:
kwargs = pickle.load(f)
output_prefix = sys.argv[3]
kwargs["num_replicates"] = num_replicates
for j, ts in enumerate(msprime.simulate(**kwargs)):
ts.dump(f"{output_prefix}_{j}.trees")