run module registry
Add some kind of registry (in the first stage maybe a list, or ordered dict). To run a custom workflow, add each stage via .add(stage, kwargs)
and finally start experiment by calling .run()
.
The run module will look like
with RunEnvironment:
for stage, kwargs in stage_dict.items():
stage(**kwargs)