Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/core/input_reader.py
View file @ ec543191
...@@ -284,7 +284,7 @@ class InputReader: ...@@ -284,7 +284,7 @@ class InputReader:
"""Returns a tf.data.Dataset object after reading, decoding, and parsing.""" """Returns a tf.data.Dataset object after reading, decoding, and parsing."""
if tfds_builder: if tfds_builder:
dataset = self._read_tfds(input_context) dataset = self._read_tfds(input_context)
elif len(self._matched_files) > 1: elif len(matched_files) > 1:
if input_context and (len(matched_files) < if input_context and (len(matched_files) <
input_context.num_input_pipelines): input_context.num_input_pipelines):
logging.warn( 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