From 3a814826878889f4a3e2541a8fd9791b9b1948cd Mon Sep 17 00:00:00 2001 From: Youngwook Kim Date: Thu, 13 Dec 2018 15:34:24 +0900 Subject: [PATCH] Fix Python 3 compatibility issue with self_generated_targets --- tensor2tensor/utils/t2t_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensor2tensor/utils/t2t_model.py b/tensor2tensor/utils/t2t_model.py index c4ec2b87e..21e3aefe0 100644 --- a/tensor2tensor/utils/t2t_model.py +++ b/tensor2tensor/utils/t2t_model.py @@ -1337,7 +1337,7 @@ def estimator_model_fn(cls, # by logits["self_generated_targets"]. tf.logging.info("Replacing targets with model-provided targets.") features["targets"] = labels = logits.pop("self_generated_targets") - assert logits.keys() == ["logits"], ( + assert list(logits.keys()) == ["logits"], ( # See "Returns" in the "top" method docstring for the expected # "logits" format when targets are generated at training time. "Expect only key 'logits' when there is 'self_generated_targets'. "