Commit bc75fa1d authored by Antoine Kaufmann's avatar Antoine Kaufmann
Browse files

experiments: better runtime log messages

parent 3745dcb4
...@@ -3,12 +3,16 @@ import asyncio ...@@ -3,12 +3,16 @@ import asyncio
import modes.experiments as exp import modes.experiments as exp
class Run(object): class Run(object):
def __init__(self, experiment, env, outpath): def __init__(self, experiment, index, env, outpath):
self.experiment = experiment self.experiment = experiment
self.index = index
self.env = env self.env = env
self.outpath = outpath self.outpath = outpath
self.output = None self.output = None
def name(self):
return self.experiment.name + '[' + str(self.index) + ']'
class Runtime(object): class Runtime(object):
def add_run(self, run): def add_run(self, run):
pass pass
...@@ -56,11 +60,11 @@ class LocalParallelRuntime(Runtime): ...@@ -56,11 +60,11 @@ class LocalParallelRuntime(Runtime):
async def do_run(self, run): async def do_run(self, run):
''' actually starts a run ''' ''' actually starts a run '''
await run.experiment.prepare(run.env, verbose=self.verbose) await run.experiment.prepare(run.env, verbose=self.verbose)
print('starting run ', run) print('starting run ', run.name())
run.output = await run.experiment.run(run.env, verbose=self.verbose) run.output = await run.experiment.run(run.env, verbose=self.verbose)
with open(run.outpath, 'w') as f: with open(run.outpath, 'w') as f:
f.write(run.output.dumps()) f.write(run.output.dumps())
print('finished run ', run) print('finished run ', run.name())
return run return run
async def wait_completion(self): async def wait_completion(self):
......
...@@ -76,6 +76,6 @@ for e in experiments: ...@@ -76,6 +76,6 @@ for e in experiments:
mkdir_if_not_exists(workdir) mkdir_if_not_exists(workdir)
env = exp.ExpEnv(args.repo, workdir) env = exp.ExpEnv(args.repo, workdir)
rt.add_run(runtime.Run(e, env, outpath)) rt.add_run(runtime.Run(e, run, env, outpath))
rt.start() rt.start()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment