dodox/Slate
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Tell Runners their current name on setup

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-02-10 19:41:24 +01:00
parent 9762e50987
commit 3d8af45c7c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
slate/slate.py
View File

@ -215,7 +215,7 @@ class Slate():
if self.consume(slurmC, 'pre_validate', True): if self.consume(slurmC, 'pre_validate', True):
Runner = self.runners[runnerName] Runner = self.runners[runnerName]
runner = Runner(self, config) runner = Runner(self, config)
runner.setup() runner.setup(wandbC['job_type'])
self._init_sweep(config) self._init_sweep(config)
self.consume(config, 'wandb') self.consume(config, 'wandb')
@ -323,7 +323,7 @@ class Slate():
**wandbC **wandbC
) as run: ) as run:
runner = Runner(self, runnerConf) runner = Runner(self, runnerConf)
runner.setup() runner.setup(wandbC['job_type'])
runner.run(run) runner.run(run)
except wandb.errors.CommError as e: except wandb.errors.CommError as e:
retry -= 1 retry -= 1
@ -363,7 +363,7 @@ class Slate():
self.deep_update(config, wandb.config) self.deep_update(config, wandb.config)
run.config = copy.deepcopy(config) run.config = copy.deepcopy(config)
runner = Runner(self, config) runner = Runner(self, config)
runner.setup() runner.setup(wandbC['job_type'])
runner.run(run) runner.run(run)
if config != {}: if config != {}: