From cc807929e6a33e06d7c90febb42b4b5ec806fd16 Mon Sep 17 00:00:00 2001
From: Ariel Kwiatkowski <ariel.j.kwiatkowski@gmail.com>
Date: Mon, 14 Mar 2022 17:40:05 +0100
Subject: [PATCH] Remove number of episodes from the search

---
 scripts/optimize/optuna_crowd.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/optimize/optuna_crowd.py b/scripts/optimize/optuna_crowd.py
index a6d886d1..9ce0be07 100644
--- a/scripts/optimize/optuna_crowd.py
+++ b/scripts/optimize/optuna_crowd.py
@@ -40,7 +40,7 @@ class Parser(BaseParser):
 def objective(trial: optuna.Trial, worker_id: int, path: str) -> float:
     # Get some parameters
     lr = trial.suggest_loguniform("lr", 1e-5, 1e-2)
-    n_episodes = trial.suggest_int("n_episodes", 1, 5)
+    n_episodes = 1
 
     steps = n_episodes * 200