TypeError: load() missing 1 required positional argument: 'Loader'

[edison046]

Excuse me, When i use training command "python Training.py" it raises TypeError: load() missing 1 required positional argument: 'Loader'. The version of musdb is 0.2.3, How to solve this problem?

Preparing MUSDB dataset! This could take a while...
ERROR - Waveunet Training - Failed after 0:00:00!
Traceback (most recent calls WITHOUT Sacred internals):
File "Training.py", line 162, in run
sup_model_path, sup_loss = optimise()
File "Training.py", line 137, in optimise
model_path = train(load_model=model_path)
File "Training.py", line 39, in train
dataset = Datasets.get_dataset(model_config, sep_input_shape, sep_output_shape, partition="train")
File "C:\Users\86135\PycharmProjects\Wave-U-Net\Wave-U-Net\Datasets.py", line 137, in get_dataset
dsd_train, dsd_test = getMUSDB(model_config["musdb_path"]) # List of (mix, acc, bass, drums, other, vocal) tuples
File "C:\Users\86135\PycharmProjects\Wave-U-Net\Wave-U-Net\Datasets.py", line 222, in getMUSDB
mus = musdb.DB(root_dir=database_path, is_wav=False)
File "C:\Users\86135\PycharmProjects\py\Wave-U-Net\lib\site-packages\musdb_init_.py", line 86, in init
self.setup = yaml.load(f)
TypeError: load() missing 1 required positional argument: 'Loader'

[alinadir44]

That happens since you are using a newer version of the PyYAML package which breaks the code.
Simply revert back to pyyaml 5.4.1 to fix

pip install pyyaml==5.4.1