def save_experiment(name, table_name, clusters, duree, mem, env, gpu):
already_existing = load_experiments(filter_eq_dct=dict(name=name))
if len(already_existing) > 0:
experiment = already_existing[0]
update_dict = dict([(key, value)
for key, value in locals().iteritems()
if key in experiment])
update_dict["clusters"] = _validate_clusters_for_saving(
clusters, experiment)
database.update("experiments", [experiment], update_dict)
else:
experiment = dict(
name=name,
table=table_name,
duree=duree,
mem=mem,
env=env,
gpu=gpu)
experiment["clusters"] = _validate_clusters_for_saving(
clusters, experiment)
experiment = database.save("experiments", experiment)
return experiment
def update_jobs(table_name, jobs, update_dict):
return database.update(table_name, jobs, update_dict)
def update_experiments(experiments, update_dict):
return database.update("experiments", experiments, update_dict)
