|
@@ -8,8 +8,11 @@ import argparse
|
|
|
|
|
|
|
|
|
def main(config_files):
|
|
|
- # This shouldn't be necessary, but...
|
|
|
- helpers.setup_environment()
|
|
|
+ # Set up buffered observer
|
|
|
+ buffered_observer = instrumentation.ObserverFactory.create({"name": "buffered"})
|
|
|
+
|
|
|
+ # Set some important environment variables and validate the GPU configuration
|
|
|
+ helpers.setup_environment([buffered_observer])
|
|
|
|
|
|
configs = [
|
|
|
pcfg.config_from_json(config_file, read_from_file=True)
|
|
@@ -24,6 +27,9 @@ def main(config_files):
|
|
|
obs = instrumentation.NamedObservable("Experiment")
|
|
|
|
|
|
experiment_container = containers.ExperimentContainer(config, outer_observable=obs)
|
|
|
+
|
|
|
+ # Replay the buffer
|
|
|
+ buffered_observer.replay_messages(obs)
|
|
|
|
|
|
with experiment_container.exception_shroud:
|
|
|
obs.log_info("Hello!")
|