An example of updating a Sumatra record

sumatra_example_update_record.py

from __future__ import unicode_literals
import numpy
import sys
import time
from sumatra.projects import load_project
from sumatra.parameters import build_parameters

def main(parameters):
    numpy.random.seed(parameters["seed"])
    distr = getattr(numpy.random, parameters["distr"])
    data = distr(size=parameters["n"])    
    output_file = "%s.dat" % parameters["sumatra_label"]
    numpy.savetxt(output_file, data)

parameter_file = sys.argv[1]
parameters = build_parameters(parameter_file)

project = load_project()    
record = project.new_record(parameters=parameters,
                            main_file=__file__,
                            reason="reason for running this simulation")
project.add_record(record)

parameters.update({"sumatra_label": record.label})
start_time = time.time()

main(parameters)

record.outcome += "Ran main function\n"
project.record_store.save(project.name, record)

do_something_else(parameters)
record.outcome += "Did something else\n"
# could also update other fields, such as stdout_stderr
project.record_store.save(project.name, record)

record.duration = time.time() - start_time
record.output_data = record.datastore.find_new_data(record.timestamp)
project.record_store.save(project.name, record)

project.save()

Thanks 👍 This was really helpful.