Function compare_structures

tests/test_sequence_packer.py:64–77

(s1, s2, rtol=1e-5, atol=1e-8)

def compare_structures(s1, s2, rtol=1e-5, atol=1e-8):
    if type(s1) != type(s2):
        return False
    if isinstance(s1, dict):
        if set(s1.keys()) != set(s2.keys()):
            return False
        return all(compare_structures(s1[k], s2[k], rtol, atol) for k in s1.keys())
    if isinstance(s1, list):
        if len(s1) != len(s2):
            return False
        return all(compare_structures(i1, i2, rtol, atol) for i1, i2 in zip(s1, s2))
    if isinstance(s1, torch.Tensor):
        return torch.allclose(s1, s2, rtol=rtol, atol=atol)
    return s1 == s2


def test_packer():