def make_dataset(tfr_pattern, mode): return dataset.DatasetSeq( tfr_pattern, job['batch_size'], job['seq_length'], job['target_length'], mode=mode)
def make_dataset(pat, mode): return dataset.DatasetSeq( tfr_pattern, job["batch_size"], job["seq_length"], job["target_length"], mode=mode, )