e4b6611cb73ef7658f028831be1aa6bd85ecbed0,tests/garage/tf/policies/test_categorical_gru_policy.py,TestCategoricalGRUPolicy,test_get_action,#TestCategoricalGRUPolicy#Any#Any#Any#,108

Before Change

        policy.reset(do_resets=None)
        obs = env.reset()[0]

        action, _ = policy.get_action(obs.flatten())
        assert env.action_space.contains(action)

        actions, _ = policy.get_actions([obs.flatten()])
        for action in actions:

After Change

        for action in actions:
            assert env.action_space.contains(action)

    @pytest.mark.parametrize("obs_dim, action_dim, hidden_dim", [
        ((1, ), 1, 4),
        ((2, ), 2, 4),
        ((1, 1), 1, 4),

Instances