increase to 90

VIDA-NYU · May 1, 2024 · 26e78e7 · 26e78e7
1 parent cc8b2ab
commit 26e78e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alpha_automl/pipeline_search/agent_environment.py b/alpha_automl/pipeline_search/agent_environment.py
@@ -30,7 +30,7 @@ def __init__(self, config: EnvContext):
         self.observation_space = Dict(
             {
                 "board": Box(
-                    0, 85, shape=(self.game.p + self.game.m,), dtype=np.uint8
+                    0, 90, shape=(self.game.p + self.game.m,), dtype=np.uint8
                 ),  # Ray env board contains pipeline and metadata
             }
         )