Skip to content

Commit

Permalink
formatting fix to test_maestro_create_input_data
Browse files Browse the repository at this point in the history
  • Loading branch information
bgenchel committed Jul 30, 2024
1 parent 415deca commit aa53a39
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/data/test_maestro.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,14 +179,14 @@ def test_maestro_invalid_tracks_over_15_min(tmpdir: str) -> None:


def test_maestro_create_input_data() -> None:
"""
A commuted metadata file is included in the repo for testing. mirdata references the metadata file to
"""
A commuted metadata file is included in the repo for testing. mirdata references the metadata file to
populate the tracklist with metadata. Since the file is commuted to only the filenames referenced here,
we only consider these when testing the metadata.
we only consider these when testing the metadata.
"""
data = create_input_data(str(MAESTRO_TEST_DATA_PATH))
assert len(data)

test_fnames = {TRAIN_TRACK_ID, VALID_TRACK_ID, TEST_TRACK_ID, GT_15M_TRACK_ID}
splits = set(map(lambda el: el[1], [d for d in data if d[0].split('.')[0] in test_fnames]))
splits = set(map(lambda el: el[1], [d for d in data if d[0].split(".")[0] in test_fnames]))
assert splits == set(["train", "validation", "test"])

0 comments on commit aa53a39

Please sign in to comment.