Reset api and run manager on each start
@@ -42,12 +42,12 @@ class JupyterAgent(object):
def __init__(self):
self.paused = True
- self.api = InternalApi()
- self.rm = RunManager(self.api, wandb.run, output=False)
def start(self):
if self.paused:
# TODO: there's an edge case where shutdown isn't called
+ self.api = InternalApi()
+ self.rm = RunManager(self.api, wandb.run, output=False)
self.api._file_stream_api = None
self.api.set_current_run_id(wandb.run.id)
self.rm.mirror_stdout_stderr()
