Commit ec543191 authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

[core][input_reader] The method should use the passed in argument other than member.

PiperOrigin-RevId: 366368802
parent dda2343a
......@@ -284,7 +284,7 @@ class InputReader:
"""Returns a tf.data.Dataset object after reading, decoding, and parsing."""
if tfds_builder:
dataset = self._read_tfds(input_context)
elif len(self._matched_files) > 1:
elif len(matched_files) > 1:
if input_context and (len(matched_files) <
input_context.num_input_pipelines):
logging.warn(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment