From d9c4b57c22fa01914c904d6fd0a7bb457c4b8e44 Mon Sep 17 00:00:00 2001
From: Maximilian Ernestus <maximilian@ernestus.de>
Date: Thu, 18 Jan 2024 14:41:39 +0100
Subject: [PATCH] Speed up trials by vectorizing environments

---
 tuning/hp_search_spaces.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tuning/hp_search_spaces.py b/tuning/hp_search_spaces.py
index d5913272a..60f0cd0ab 100644
--- a/tuning/hp_search_spaces.py
+++ b/tuning/hp_search_spaces.py
@@ -76,7 +76,7 @@ def __call__(
         suggest_named_configs=lambda _: ["reward.reward_ensemble"],
         suggest_config_updates=lambda trial: {
             "seed": trial.number,
-            "environment": {"num_vec": 1},
+            "environment": {"num_vec": 8},
             "total_timesteps": 2e7,
             "total_comparisons": 1000,
             "active_selection": True,