error when running make_test_train with --chordwise

[antoniomechas]

Hello,
When running "python make_test_train.py --chordwise I get the following error:

python make_test_train.py --chordwise
data/composers/chordwise/piano_solo/note_range62/sample_freq4
data/composers/chordwise/piano_solo/note_range62/sample_freq4/mozart-maestro
Chordwise encoding: removing duration marks and expanding rests
Traceback (most recent call last):
  File "make_test_train.py", line 216, in <module>
    main(SOURCE, TARGET_TRAIN, TARGET_TEST, composer, args.tt_split, args.chordwise, args.sample, args.no_wait)
  File "make_test_train.py", line 161, in main
    remove_duration(TARGET_TRAIN)
  File "make_test_train.py", line 50, in remove_duration
    for file in tdqm(os.listdir(DIR)):
NameError: name 'tdqm' is not defined

I replaced the "tdqm" by "tqdm".

Then I got the error:

data/composers/chordwise/piano_solo/note_range62/sample_freq4
data/composers/chordwise/piano_solo/note_range62/sample_freq4/mozart-maestro
Chordwise encoding: removing duration marks and expanding rests
  0%|                                                                                                  | 0/34 [00:00<?, ?it/s]
Traceback (most recent call last):
  File "make_test_train.py", line 216, in <module>
    main(SOURCE, TARGET_TRAIN, TARGET_TEST, composer, args.tt_split, args.chordwise, args.sample, args.no_wait)
  File "make_test_train.py", line 161, in main
    remove_duration(TARGET_TRAIN)
  File "make_test_train.py", line 59, in remove_duration
    f.write(" ".join(temp[i*piece_len:(i+1)*piece_len]))
NameError: name 'piece_len' is not defined

I added the line:
piece_len=len(temp)//12
just before the for in line 56.

Then it worked fine... hope those modifications are right.
Antonio.