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

chess-gpt-eval-contrastive/mamba_module.py

@@ -213,7 +213,7 @@ class MambaPlayer:
 
     def update_activations(self, result):
         for layer_idx in self.activations_sum:
-            if "result" == "reset":
+            if result == "reset":
                 self.activations_sum[layer_idx] = {bucket: {"won": np.zeros((1, self.max_seq_len, self.model.config.d_model)),
                                                     "lost": np.zeros((1, self.max_seq_len, self.model.config.d_model)),
                                                     "current": np.zeros((1, self.max_seq_len, self.model.config.d_model))}