Update test/exceptions/test_errors.py

Co-authored-by: Max Balandat <[email protected]>
pytorch · Sep 15, 2024 · a44d242 · a44d242
1 parent d14baee
commit a44d242
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/exceptions/test_errors.py b/test/exceptions/test_errors.py
@@ -54,5 +54,5 @@ def test_OptimizationTimeoutError(self):
     def test_OptimizationGradientError(self):
         error = OptimizationGradientError("message", current_x=np.array([1.0]))
         self.assertTrue(np.array_equal(error.current_x, np.array([1.0])))
-        with self.assertRaises(OptimizationGradientError):
+        with self.assertRaisesRegex(OptimizationGradientError, "message"):
             raise error