From bc20bfcde2f4980654f72f34a87a57074f8d16c3 Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Tue, 26 Nov 2024 22:00:21 -0500
Subject: [PATCH] fix autotune

---
 pokemonred_puffer/train.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py
index c11a688..43b4ee6 100644
--- a/pokemonred_puffer/train.py
+++ b/pokemonred_puffer/train.py
@@ -250,7 +250,7 @@ def autotune(
             "-r",
             help="Reward module to use in rewards",
         ),
-    ] = "baseline.BaselineRewardEnv",
+    ] = "baseline.ObjectRewardRequiredEventsMapIds",
     wrappers_name: Annotated[
         str,
         typer.Option(
@@ -258,9 +258,11 @@ def autotune(
             "-w",
             help="Wrappers to use _in order of instantion_",
         ),
-    ] = "baseline",
+    ] = "empty",
     rom_path: Path = "red.gb",
 ):
+    config = load_from_config(config, False)
+    config.vectorization = "multiprocessing"
     config, env_creator = setup(
         config=config,
         debug=False,