From 3d8af45c7c2579da0e92b6ef72abbbbbc3eae593 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Sat, 10 Feb 2024 19:41:24 +0100
Subject: [PATCH] Tell Runners their current name on setup

---
 slate/slate.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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