subhdotsol commited on
Commit
061383a
·
1 Parent(s): b8bc48b

feat(environment): add set_reward_computer() injection method for Person 2

Browse files
Files changed (1) hide show
  1. server/environment.py +4 -0
server/environment.py CHANGED
@@ -23,3 +23,7 @@ class RedTeamEnvironment:
23
 
24
  self.reward_computer: Optional[Callable] = None
25
  self.llm_pipeline: Optional[Callable] = None
 
 
 
 
 
23
 
24
  self.reward_computer: Optional[Callable] = None
25
  self.llm_pipeline: Optional[Callable] = None
26
+
27
+ def set_reward_computer(self, compute_fn: Callable) -> None:
28
+ self.reward_computer = compute_fn
29
+ logger.info("Reward computer wired up (Person 2)")