subhdotsol commited on
Commit
348c725
·
1 Parent(s): e4d4c65

feat(environment): call reward_computer.reset() in reset() to clear novelty scorer

Browse files
Files changed (1) hide show
  1. server/environment.py +3 -0
server/environment.py CHANGED
@@ -39,6 +39,9 @@ class RedTeamEnvironment:
39
  self.is_active = True
40
  self.created_at = datetime.utcnow()
41
 
 
 
 
42
  logger.info(f"Episode started: {self.episode_id}")
43
 
44
  return RedTeamObservation(
 
39
  self.is_active = True
40
  self.created_at = datetime.utcnow()
41
 
42
+ if self.reward_computer and hasattr(self.reward_computer, "reset"):
43
+ self.reward_computer.reset()
44
+
45
  logger.info(f"Episode started: {self.episode_id}")
46
 
47
  return RedTeamObservation(