Fix pytype error

PiperOrigin-RevId: 588730108 Change-Id: I1bd662c95e9e02d2b7dec8140716c77af014c33f
google-deepmind · Dec 7, 2023 · c8091ae · c8091ae
1 parent 5918eea
commit c8091ae
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/cyberball/components/ball_status.py b/examples/cyberball/components/ball_status.py
@@ -64,7 +64,7 @@ def partial_state(
     """Return a player-specific view of the ball's state."""
     return self._partial_states[player_name]
 
-  def update(self) -> str:
+  def update(self) -> None:
     self._state = '\n'
     self._partial_states = {name: '' for name in self._player_names}
     per_player_prompt = {}
@@ -99,5 +99,3 @@ def update(self) -> str:
         'per player prompts': per_player_prompt,
     }
     self._history.append(update_log)
-
-    return self._state