class_docstr: Models a simple one variable(y = wx) neural network, and a MSE loss function.
class_docstr_tok: ["Models","a","simple","one","variable","(","y","=","wx",")","neural","network",",","and","a","MSE","loss","function","."]
code: def build_validation_data_loader(self) -> keras.InputData:
      dataset = make_one_var_tf_dataset_loader(
          self.context.get_hparams(), self.my_batch_size
      )
      dataset = self.context.wrap_dataset(dataset)
      return dataset
  
file_path: determined/harness/tests/experiment/fixtures/tf_keras_one_var_model.py
imports: from typing import Any, Dict, List, cast
  import numpy
  import tensorflow
  from tensorflow.keras.layers import Dense
  from tensorflow.keras.losses import mean_squared_error
  from tensorflow.keras.models import Sequential
  from tensorflow.keras.optimizers.legacy import SGD
  from determined import keras
  from tests.experiment.fixtures import keras_cb_checker
label_desc: Use image node_id 3 for calling the OneVarTrial obj's underlying member method code with example usage: obj.build_validation_data_loader() and returns: dataset