Merge pull request #76 from opentensor/mock-blacklist-fix

ensure mock reward model has new blacklist functions or crashes
opentensor · Nov 20, 2023 · 7709305 · 7709305
2 parents f104e25 + 4eb41fa
commit 7709305
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/prompting/validators/mock.py b/prompting/validators/mock.py
@@ -60,6 +60,12 @@ def __init__(self, mock_name: str = "MockReward"):
         self.question_blacklist = []
         self.answer_blacklist = []
 
+    def add(self, texts: List[str]):
+        pass
+
+    def set_counter_to_half(self):
+        pass
+
     def apply(self, prompt: str, completion: List[str], name: str) -> torch.FloatTensor:
         mock_reward = torch.tensor([1 for _ in completion], dtype=torch.float32)
         return mock_reward, {}