code: def check_obs(obs, grid_size):
      assert len(obs) == 2, "two players"
      for key, player_obs in obs.items():
          assert player_obs.shape == (grid_size, grid_size, 4)
          assert (
              player_obs[..., 0].sum() == 1.0
          ), f"observe 1 player red in grid: {player_obs[..., 0]}"
          assert (
              player_obs[..., 1].sum() == 1.0
          ), f"observe 1 player blue in grid: {player_obs[..., 1]}"
          assert (
              player_obs[..., 2:].sum() == 1.0
          ), f"observe 1 coin in grid: {player_obs[..., 0]}"
  
imports: import random
  import numpy
  from ray.rllib.examples.env.coin_game_non_vectorized_env import CoinGame, AsymCoinGame
label_desc: Use image node_id 4 for calling a global function with example usage: check_obs(obs, grid_size) without return types