2a26887b42ac2d514fc7cd54187402581a054275,ml-agents/mlagents/trainers/tests/test_agent_processor.py,,test_agentprocessor,#Any#,31

Before Change

        stats_reporter=StatsReporter("testcat"),
    )

    fake_action_outputs = {
        "action": ActionTuple(continuous=np.array([[0.1], [0.1]], dtype=np.float32)),
        "entropy": np.array([1.0], dtype=np.float32),
        "learning_rate": 1.0,
        "log_probs": LogProbsTuple(
            continuous=np.array([[0.1], [0.1]], dtype=np.float32)
        ),
    }
    mock_decision_steps, mock_terminal_steps = mb.create_mock_steps(
        num_agents=2,
        observation_specs=create_observation_specs_with_shapes(
            [(8,)] + num_vis_obs * [(84, 84, 3)]

After Change

    assert len(trajectory.steps) == 5
    // Make sure ungrouped agents don"t have team obs
    for step in trajectory.steps:
        assert len(step.group_status) == 0

    // Assert that the AgentProcessor is empty
    assert len(processor._experience_buffers[0]) == 0

Instances