2f71f723d1d9d417a70f832a0c9178f75da75e18,ml-agents/mlagents/trainers/tests/torch/test_distributions.py,,test_gaussian_distribution,#,15

Before Change

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
    for prob in log_prob.flatten():
        assert prob == pytest.approx(-2, abs=0.1)

After Change

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
    for prob in log_prob.flatten().tolist():
        assert prob == pytest.approx(-2, abs=0.1)

Instances