Test failures print message in assertNormsClose

Summary: Restore assertNormsClose's printing of its message on failure which I broke in D26233419 (cd9786e787).

Reviewed By: nikhilaravi

Differential Revision: D28799743

fbshipit-source-id: e7a24b2558b68991c731bbd55fb3ca6c1df98f69
This commit is contained in:
Jeremy Reizenstein 2021-06-03 18:29:42 -07:00 committed by Facebook GitHub Bot
parent 070ec550d3
commit 36b451a49b

View File

@ -120,7 +120,7 @@ class TestCaseMixin(unittest.TestCase):
# all(norm_fn(input - other) <= atol + rtol * norm_fn(other)).
self.assertClose(
diff + other_, other_, rtol=rtol, atol=atol, equal_nan=equal_nan
diff + other_, other_, rtol=rtol, atol=atol, equal_nan=equal_nan, msg=msg
)
def assertClose(