d7177951755d388cc929ad559d7dbde17a019b7c,tests/garage/torch/policies/test_categorical_cnn_policy.py,TestCategoricalCNNPolicy,test_is_pickleable,#TestCategoricalCNNPolicy#Any#Any#Any#Any#,105

Before Change

                                      strides=strides,
                                      hidden_sizes=hidden_sizes)
        env.reset()
        obs, _, _, _ = env.step(1)

        output_action_1, _ = policy.get_action(obs)

After Change

    def test_is_pickleable(self, hidden_channels, kernel_sizes, strides,
                           hidden_sizes):
        Test if policy is pickable.
        env = GymEnv(DummyDiscretePixelEnv(), is_image=True)
        env = self._initialize_obs_env(env)
        policy = CategoricalCNNPolicy(env=env,
                                      kernel_sizes=kernel_sizes,
                                      hidden_channels=hidden_channels,
                                      strides=strides,
                                      hidden_sizes=hidden_sizes)
        env.reset()
        obs = env.step(1).observation

        output_action_1, _ = policy.get_action(obs)

        p = cloudpickle.dumps(policy)

Instances