No public description

PiperOrigin-RevId: 553708377

official/projects/yt8m/tasks/yt8m_task.py

@@ -19,13 +19,13 @@ from absl import logging
 import tensorflow as tf
 
 from official.core import base_task
-from official.core import input_reader
 from official.core import task_factory
 from official.modeling import tf_utils
 from official.projects.yt8m.configs import yt8m as yt8m_cfg
 from official.projects.yt8m.dataloaders import yt8m_input
 from official.projects.yt8m.eval_utils import eval_util
 from official.projects.yt8m.modeling import yt8m_model
+from official.vision.dataloaders.google import input_reader
 
 
 @task_factory.register_task_cls(yt8m_cfg.YT8MTask)

official/vision/dataloaders/input_reader.py

@@ -23,6 +23,9 @@ from official.core import config_definitions as cfg
 from official.core import input_reader
 
 
+InputReader = input_reader.InputReader
+
+
 def build_weighted_sampling_combine_fn(
     weights: Mapping[Any, Any], stop_on_empty_dataset=True
 ) -> Callable[[tf.data.Dataset], tf.data.Dataset]: