Update chess-gpt-eval-contrastive/mamba_module.py

chess-gpt-eval-contrastive/mamba_module.py

@@ -160,6 +160,10 @@ class MambaPlayer:
     def get_config(self) -> dict:
         return {"model": self.model_name}
 
+    def update_activations(self, result):
+        for layer_idx in self.activations:
+            self.activations[layer_idx][result].append(self.activations[layer_idx]["current"])
+
     def save_activations(self, path):
         activations_sum = {}
         activations_count = {}