fix(pu): fix test_mcts_sampled_ctree

opendilab · Oct 30, 2023 · 668ab4c · 668ab4c
1 parent 4ac9c8f
commit 668ab4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lzero/mcts/tests/test_mcts_sampled_ctree.py b/lzero/mcts/tests/test_mcts_sampled_ctree.py
@@ -29,7 +29,7 @@ def initial_inference(self, observation):
         reward_hidden_state_state = (torch.zeros(size=(1, batch_size, 16)), torch.zeros(size=(1, batch_size, 16)))
 
         output = {
-            'searched_value': value,
+            'value': value,
             'value_prefix': value_prefix,
             'policy_logits': policy_logits,
             'latent_state': latent_state,
@@ -48,7 +48,7 @@ def recurrent_inference(self, hidden_states, reward_hidden_states, actions):
         # policy_logits = torch.zeros(size=(batch_size, self.action_num))
 
         output = {
-            'searched_value': value,
+            'value': value,
             'value_prefix': value_prefix,
             'policy_logits': policy_logits,
             'latent_state': latent_state,