[BUG] Tests failing due to retiring of torch.testing.assert_allclose

[sosmond]

Describe the bug

Due to this change, torch.testing.assert_allclose has been removed which is causing unit tests to fail.

Output from a recent CI run

FAILED test/test_rb.py::test_prototype_prb[device0-True-pk] - AttributeError: module 'torch.testing' has no attribute 'assert_allclose'
FAILED test/test_rb.py::test_prototype_prb[device0-True-td_error] - AttributeError: module 'torch.testing' has no attribute 'assert_allclose'
FAILED test/test_rb.py::test_prototype_prb[device0-False-pk] - AttributeError: module 'torch.testing' has no attribute 'assert_allclose'
FAILED test/test_rb.py::test_prototype_prb[device0-False-td_error] - AttributeError: module 'torch.testing' has no attribute 'assert_allclose'

To Reproduce

Run pytest with pytorch nightly build installed.

Reason and Possible fixes

torch.testing.assert_close should be a drop-in replacement; reference

Checklist

• I have checked that there is no similar issue in the repo (required)
• I have read the documentation (required)
• I have provided a minimal working example to reproduce the bug (required)

[vmoens]

Solved as of #644