Skip to content

Commit 14ec065

Browse files
committed
Adding missing env_layout_seed info to SB3 instincts
1 parent 2caa3ad commit 14ec065

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

aintelope/agents/sb3_handwritten_rules_expert.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def should_override(
9292
self,
9393
deterministic: bool = False, # This is set only during evaluation, not training and the meaning is that the agent is greedy - it takes the best action. It does NOT mean that the action is always same.
9494
step: int = 0,
95+
env_layout_seed: int = 0,
9596
episode: int = 0,
9697
pipeline_cycle: int = 0,
9798
test_mode: bool = False,
@@ -184,6 +185,7 @@ def get_action(
184185
observation=None,
185186
info: dict = {},
186187
step: int = 0,
188+
env_layout_seed: int = 0,
187189
episode: int = 0,
188190
pipeline_cycle: int = 0,
189191
test_mode: bool = False,

0 commit comments

Comments
 (0)