32ab09487234c61244b2d11fd0bf0a13a96969d1,tests/test_pm_imagenet.py,,equal,#,17

Before Change

def equal(x,y):
    return torch.le(torch.dist(x, y), 1e-6)

@pytest.mark.parametrize("model_name, pretrained", pm_args)
def test_pm_imagenet(model_name, pretrained):
    print("test_pm_imagenet("{}")".format(model_name))

After Change

def equal(x,y):
    return torch.all(torch.lt(torch.abs(torch.add(x, -y)), 1e-12))

@pytest.mark.parametrize("model_name, pretrained", pm_args)
def test_pm_imagenet(model_name, pretrained):
    print("test_pm_imagenet("{}")".format(model_name))

Instances