_assert_utils.py 322 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
"""This is a temporary module and should be removed as soon as torch.testing.assert_equal is supported."""
# TODO: remove this as soon torch.testing.assert_equal is supported

import functools

import torch.testing

__all__ = ["assert_equal"]


assert_equal = functools.partial(torch.testing.assert_close, rtol=0, atol=0)