Fix runner eval test (#170)

* Fix runner eval test * double negative
ucl-dark · Oct 25, 2023 · 00a6b6f · 00a6b6f
1 parent 7ef09e1
commit 00a6b6f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pax/runners/runner_eval.py b/pax/runners/runner_eval.py
@@ -305,7 +305,10 @@ def run_loop(self, env, env_params, agents, num_episodes, watchers):
             self.model_path2 is not None and self.run_path2 is not None
         )
 
-        if watchers and not self.args.wandb.mode == "offline":
+        if watchers and self.args.wandb.mode not in [
+            "offline",
+            "disabled",
+        ]:
             wandb.restore(
                 name=self.model_path, run_path=self.run_path, root=os.getcwd()
             )