code: def sample_data():
      # spaces
      obs_space = akro.Box(
          low=1, high=10, shape=(4, 3, 2), dtype=np.float32
      )
      act_space = akro.Discrete(5)
      env_spec = EnvSpec(obs_space, act_space)
  
      # generate data
      next_obs = obs_space.sample()
      act = act_space.sample()
      rew = 10.0
      step_type = StepType.FIRST
  
      # env_infos
      env_infos = dict()
      env_infos["goal"] = np.array([[1, 1]])
      env_infos["TimeLimit.truncated"] = step_type == StepType.TIMEOUT
  
      return {
          "env_spec": env_spec,
          "observation": next_obs,
          "action": act,
          "reward": rew,
          "env_info": env_infos,
          "step_type": step_type,
      }
  
file_path: garage/tests/garage/test_environment.py
imports: import akro
  import cloudpickle
  import numpy
  import pytest
  from garage import EnvSpec, EnvStep, StepType
label_desc: Use image node_id 2 for calling a global function with example usage: sample_data() and returns: dict