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

chess-gpt-eval-contrastive/mamba_module.py

@@ -174,7 +174,7 @@ class MambaPlayer:
         return {"model": self.model_name}
 
     def update_activations(self, result):
-        for layer_idx in self.activations:
+        for layer_idx in self.activations_sum:
             self.activations_sum[layer_idx][result] += self.activations[layer_idx]["current"]
             self.activations_count[layer_idx][result] += 1