docstr: Test get_action function with akro.Image observation space.
docstr_tok: ["Test","get_action","function","with","akro.Image","observation","space","."]
code: def test_get_action_img_obs(
      self, hidden_channels, kernel_sizes, strides, hidden_sizes
  ):
      """Test get_action function with akro.Image observation space."""
      env = GymEnv(DummyDiscretePixelEnv(), is_image=True)
      policy = CategoricalCNNPolicy(
          env_spec=env.spec,
          image_format="NHWC",
          kernel_sizes=kernel_sizes,
          hidden_channels=hidden_channels,
          strides=strides,
          hidden_sizes=hidden_sizes,
      )
      env.reset()
      obs = env.step(1).observation
  
      action, _ = policy.get_action(obs)
      assert env.action_space.contains(action)
  
file_path: garage/tests/garage/torch/policies/test_categorical_cnn_policy.py
imports: import cloudpickle
  import pytest
  import torch
  from garage.envs import GymEnv
  from garage.torch.policies import CategoricalCNNPolicy
  from tests.fixtures.envs.dummy import DummyDictEnv, DummyDiscretePixelEnv
label_desc: Use image node_id 2 for calling the TestCategoricalCNNPolicy obj's underlying member method code with example usage: obj.test_get_action_img_obs(hidden_channels, kernel_sizes, strides, hidden_sizes) without return types