Move tensorflow/tsl/protobuf to xla/tsl/protobuf

PiperOrigin-RevId: 644440378
tensorflow · Jun 20, 2024 · 1f06e44 · 1f06e44
1 parent f94b329
commit 1f06e44
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tfx/components/infra_validator/model_server_clients/tensorflow_serving_client.py b/tfx/components/infra_validator/model_server_clients/tensorflow_serving_client.py
@@ -37,7 +37,8 @@
                   'the TF version doesn\'t match with the TF Serving version. '
                   'We will try importing again with a workaround:%s', err)
   from tensorflow.core.protobuf import error_codes_pb2 as old_error_codes_pb2
-  from tensorflow.tsl.protobuf import error_codes_pb2 as new_error_codes_pb2
+  from tensorflow.compiler.xla.tsl.protobuf import error_codes_pb2 as new_error_codes_pb2
+
   old_error_codes_pb2._CODE = new_error_codes_pb2._CODE  # pylint: disable=protected-access  # pytype: disable=module-attr
 
   # Retry.