def METHOD_NAME(lbann):
    """Construct Protobuf message for Python data reader.
    The Python data reader will import the current Python file to
    access the sample access functions.
    Args:
        lbann (module): Module for LBANN Python frontend
    """
    # Note: The training data reader should be removed when
    # https://github.com/LLNL/lbann/issues/1098 is resolved.
    message = lbann.reader_pb2.DataReader()
    message.reader.extend(
        [
            tools.create_python_data_reader(
                lbann, current_file, "get_sample", "num_samples", "sample_dims", "train"
            )
        ]
    )
    message.reader.extend(
        [
            tools.create_python_data_reader(
                lbann, current_file, "get_sample", "num_samples", "sample_dims", "test"
            )
        ]
    )
    return message