Address review.

Author: ysiraichi

test/test_operations.py

@@ -2460,7 +2460,7 @@ def test_add_broadcast_error(self):
 
   def test_construct_large_tensor_raises_error(self):
     with self.assertRaisesRegex(RuntimeError,
-                                r"Out of memory allocating \d* bytes"):
+                                r"Out of memory allocating \d+ bytes"):
       # When eager-mode is enabled, OOM is triggered here.
       a = torch.rand(1024, 1024, 1024, 1024, 1024, device=torch_xla.device())
       b = a.sum()