From 64d839bb754b104e151bb49bb4ec46dbe690745d Mon Sep 17 00:00:00 2001 From: bhack Date: Fri, 15 May 2020 18:21:51 +0200 Subject: [PATCH] Fix lint and improve readibility --- tensorflow/python/ops/map_fn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/ops/map_fn.py b/tensorflow/python/ops/map_fn.py index b98b4ad10bc..40f8edfcdd1 100644 --- a/tensorflow/python/ops/map_fn.py +++ b/tensorflow/python/ops/map_fn.py @@ -479,7 +479,8 @@ def map_fn(fn, elems_value_flat = _elems_value_batchable_to_flat(elems_value_batchable, elems_flat_signature) elems_value = elems_unflatten(elems_value_flat) - autographed_fn = autograph.tf_convert(fn, autograph_ctx.control_status_ctx()) + ag_ctx = autograph_ctx.control_status_ctx() + autographed_fn = autograph.tf_convert(fn, ag_ctx) result_value = autographed_fn(elems_value) nest.assert_same_structure(fn_output_signature or elems, result_value) result_value_flat = nest.flatten(result_value)