d658c8b1a2627beb7f9cc6e84b3bdcf227fc3b6d,tests/ignite/metrics/test_root_mean_squared_error.py,,test_compute,#,19

Before Change

    y = torch.zeros(2)
    rmse.update((y_pred, y))
    assert isinstance(rmse.compute(), float)
    assert rmse.compute() == 3.0


def _test_distrib_integration(device, tol=1e-6):
    import numpy as np

After Change

        return test_cases

    for _ in range(5):
        // check multiple random inputs as random exact occurencies are rare
        test_cases = get_test_cases()
        for y_pred, y, batch_size in test_cases:
            _test(y_pred, y, batch_size)


def _test_distrib_integration(device, tol=1e-6):

    from ignite.engine import Engine

Instances