Fix bug in assertRaises NotImplemented handling when no exception is thrown (#54126)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54126
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Test Plan: Imported from OSS
Reviewed By: agolynski, mruberry
Differential Revision: D27109510
Pulled By: ezyang
fbshipit-source-id: ba5a4de85ca00f81724f3d4e645797e8f32aa3b1