def performances_input_transform(self, tmp_dir):
del tmp_dir
return dict(
(split_name,
datagen_beam.ReadNoteSequencesFromTFRecord(tfrecord_path))
for split_name, tfrecord_path in MAESTRO_TFRECORD_PATHS.items())
def _maestro_input_transform():
from magenta.models.score2perf import datagen_beam # pylint: disable=g-import-not-at-top,import-outside-toplevel
return dict(
(split_name, datagen_beam.ReadNoteSequencesFromTFRecord(tfrecord_path))
for split_name, tfrecord_path in MAESTRO_TFRECORD_PATHS.items())
