OpenAI Gym GoogleColabで使いたい

Question

OpenAI Gymに似た、PettingZooと呼ばれるAPIを用いて、シミュレーションをGoogle Colabで行いたいと思っています。
https://www.pettingzoo.ml/#

!pip install pettingzoo[butterfly]
を実行したあと、チュートリアルに書かれている以下のコードを実行したのですが、
name 'policy' is not defined
というエラーが出てしまいました。どのようにすれば使えるようになるのでしょうか。

  from pettingzoo.butterfly import knights_archers_zombies_v7
  env = knights_archers_zombies_v7.env()
  env.reset()
  for agent in env.agent_iter():
      observation, reward, done, info = env.last()
      action = policy(observation, agent)
      env.step(action)

Answer 1

policyは自分で定義する必要がありそうです。