Fix code discrepancies encountered during the open-sourcing process - #6

Fix code discrepancies encountered during the open-sourcing process#6

hyp1231 wants to merge 13 commits into

google-deepmind:maingoogle-deepmind/action_piece:mainfrom

hyp1231:mainhyp1231/action_piece:mainCopy head branch name to clipboard