Remove the copybara replacement of the error message.

PiperOrigin-RevId: 565170171
google-parfait · Sep 13, 2023 · 374d186 · 374d186
1 parent 4a28765
commit 374d186
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow_federated/python/tensorflow_libs/variable_utils_test.py b/tensorflow_federated/python/tensorflow_libs/variable_utils_test.py
@@ -226,7 +226,8 @@ def test_scalar_getitem_fails(self, value):
     value = tf.convert_to_tensor(value)
     tensor_variable = variable_utils.TensorVariable(value)
     with self.assertRaisesRegex(
-        tf.errors.InvalidArgumentError, 'Index out of range'
+        tf.errors.InvalidArgumentError,
+        'Attempting to slice scalar input',
     ):
       _ = tensor_variable[0]