Spaces:
Sleeping
Sleeping
Commit ·
061383a
1
Parent(s): b8bc48b
feat(environment): add set_reward_computer() injection method for Person 2
Browse files- 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)")
|