Merge pull request #92 from KazukiOhta:patch-1

PiperOrigin-RevId: 642908290
google-deepmind · Jun 13, 2024 · 9fb7339 · 9fb7339
2 parents be1dbb1 + 4379f2c
commit 9fb7339
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mctx/_src/tree.py b/mctx/_src/tree.py
@@ -42,7 +42,7 @@ class Tree(Generic[T]):
     node.
   children_index: `[B, N, num_actions]` the node index of the children for each
     action.
-  children_prior_logits: `[B, N, Anum_actions` the action prior logits of each
+  children_prior_logits: `[B, N, num_actions]` the action prior logits of each
     node.
   children_visits: `[B, N, num_actions]` the visit counts for children for
     each action.