From 3d72579d4ff68f30e723a626b3c7acbc59514674 Mon Sep 17 00:00:00 2001 From: arthurdrake1 Date: Sat, 16 Aug 2025 21:52:35 +0200 Subject: [PATCH] add state reinitialization to reset function --- engibench/problems/beams2d/v0.py | 1 + 1 file changed, 1 insertion(+) diff --git a/engibench/problems/beams2d/v0.py b/engibench/problems/beams2d/v0.py index a5f9f4d7..6803d168 100644 --- a/engibench/problems/beams2d/v0.py +++ b/engibench/problems/beams2d/v0.py @@ -292,6 +292,7 @@ def reset(self, seed: int | None = None, **kwargs) -> None: **kwargs: Additional keyword arguments. """ super().reset(seed, **kwargs) + self.__st = State() def render(self, design: np.ndarray, *, open_window: bool = False) -> Any: """Renders the design in a human-readable format.