psweep | Loop like a pro , make parameter studies | Development Tools library

import random
import psweep as ps


def func(pset):
    return {'result': random.random() * pset['a'] * pset['b']}


if __name__ == '__main__':
    a = ps.plist('a', [1,2,3])
    b = ps.plist('b', [77,88])
    params = ps.pgrid(a,b)
    df = ps.run_l  

params = [{'a': 1, 'b': 'lala'},  # pset 1
          {'a': 2, 'b': 'zzz'},   # pset 2
          ...                     # ...
         ]

{'a': 1, 'b': 'lala'}

{'result': 1.234}

{'a': 1, 'b': 'lala', 'result': 1.234}

>>> from itertools im  

# templates/machines/cluster/info.yaml
---
subcmd: sbatch

#SBATCH --time 00:20:00
#SBATCH -o out.job
#SBATCH -J foo_${_pset_seq}_${_pset_id}
#SBATCH -p bar
#SBATCH -A baz

# Because we use Python's string.Template, we need to escape the dollar char