Add cast preprocessor and add tasks for inference prompts for dedup…

…lication project. PiperOrigin-RevId: 373369846
tensorflow · May 12, 2021 · a2f59cb · a2f59cb
1 parent da90793
commit a2f59cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mesh_tensorflow/transformer/utils.py b/mesh_tensorflow/transformer/utils.py
@@ -1667,7 +1667,7 @@ def score_with_estimator(estimator, input_fn, eval_checkpoint_step, model_dir,
   Returns:
     a list of floats
   """
-  checkpoint_path, = get_checkpoint_iterator(eval_checkpoint_step, model_dir)
+  checkpoint_path = get_checkpoint_iterator(eval_checkpoint_step, model_dir)
 
   result_iter = estimator.predict(input_fn, checkpoint_path=checkpoint_path)
   # TODO(dei): This code is not well-designed for large-scale scoring, where the