Call run_returning_status instead of run

PiperOrigin-RevId: 458052236
tensorflow · Jun 29, 2022 · 8f9085a · 8f9085a
1 parent d8a919b
commit 8f9085a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_serving/experimental/example/half_plus_two_with_rpop.py b/tensorflow_serving/experimental/example/half_plus_two_with_rpop.py
@@ -69,7 +69,7 @@ def _generate_saved_model_for_half_plus_two(export_dir, target_address,
       output_tensor_aliases = tf.constant(["y"])
       output_types = [tf.float32]
 
-      results = remote_predict_ops.run(
+      results = remote_predict_ops.run_returning_status(
           input_tensor_aliases,
           input_tensors,
           output_tensor_aliases,