code: def test_agent_infos_batch_mismatch_eps(eps_data):
      with pytest.raises(
          ValueError,
          match="Entry 'hidden' in agent_infos has batch size 141",
      ):
          eps_data["agent_infos"]["hidden"] = eps_data["agent_infos"][
              "hidden"
          ][:-1]
          t = EpisodeBatch(**eps_data)
          del t
  
imports: import akro
  import gym.spaces
  import numpy
  import pytest
  from garage import EnvSpec, EnvStep, EpisodeBatch, StepType, TimeStep, TimeStepBatch
label_desc: Use image node_id 16 for calling a global function with example usage: test_agent_infos_batch_mismatch_eps(eps_data) without return types