Skip to content

Commit f686acf

Browse files
committed
Issue #144: Update composed reward
1 parent d861f97 commit f686acf

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

src/bsk_rl/data/composition.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,12 @@ def reset_pre_sim_init(self) -> None:
157157
for rewarder in self.rewarders:
158158
rewarder.reset_pre_sim_init()
159159

160+
def reset_during_sim_init(self) -> None:
161+
"""Handle resetting for all rewarders."""
162+
super().reset_during_sim_init()
163+
for rewarder in self.rewarders:
164+
rewarder.reset_during_sim_init()
165+
160166
def reset_post_sim_init(self) -> None:
161167
"""Handle resetting for all rewarders."""
162168
super().reset_post_sim_init()

tests/unittest/data/test_composition.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def test_pass_data(self):
106106
"function",
107107
[
108108
"reset_pre_sim_init",
109+
"reset_during_sim_init",
109110
"reset_post_sim_init",
110111
"reset_overwrite_previous",
111112
],

0 commit comments

Comments
 (0)