From 22a990b1c3201b32a9c679dfb922043d692d8959 Mon Sep 17 00:00:00 2001 From: Justin DuJardin Date: Mon, 10 Oct 2022 14:23:52 -0700 Subject: [PATCH] feat(swarm): increase solver state size to support longer problems (#82) - the env state has to keep the entire move history in its output (to penalize reentrant states, etc.) and the 768 length can be too short on long problems --- libraries/mathy_python/mathy/solver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libraries/mathy_python/mathy/solver.py b/libraries/mathy_python/mathy/solver.py index 5be3e5e4..30d4d539 100644 --- a/libraries/mathy_python/mathy/solver.py +++ b/libraries/mathy_python/mathy/solver.py @@ -147,7 +147,7 @@ def __init__( def get_state(self) -> np.ndarray: assert self._env.state is not None, "env required to get_state" - return self._env.state.to_np(768) + return self._env.state.to_np(2048) def set_state(self, state: np.ndarray): assert self._env is not None, "env required to set_state"