Skip to content

Commit db81c21

Browse files
committed
[MINOR] Avoid bad RLlib logs on training restart
1 parent 5172a46 commit db81c21

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/bsk_rl/utils/rllib/callbacks.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,7 @@ def __init__(self, episode_data_callback=None, satellite_data_callback=None):
185185
self.metrics_logger = None
186186
self.env: "GeneralSatelliteTasking"
187187
self.satellites: list["Satellite"]
188+
self.logs_to_skip = 3 # Hacky way to avoid a log on initial reset
188189

189190
def set_metrics_logger(self, metrics_logger):
190191
"""Set the metrics logger for this environment."""
@@ -216,7 +217,10 @@ def log_data_on_reset(self):
216217

217218
def reset(self, **kwargs):
218219
"""Log data before resetting the environment."""
219-
self.log_data_on_reset()
220+
if self.logs_to_skip > 0:
221+
self.logs_to_skip -= 1
222+
else:
223+
self.log_data_on_reset()
220224
return self.env.reset(**kwargs)
221225

222226

0 commit comments

Comments
 (0)