Update the arg names.
Update the arg names.

PiperOrigin-RevId: 272225488
Change-Id: Id2ccab89e917ddd6dfded2959f853b16fd679955

deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel,deepmind/open_spiel