Loader positional argument missing - Colab notebook

[fromglow]

Hello,
When I run the "generate dance" cell in the colab notebook, I get an error that says "Traceback (most recent call last):
File "inference/generate.py", line 76, in
exp_opt = yaml.load(open(str(checkpoint_dir)+"/hparams.yaml","r").read())
TypeError: load() missing 1 required positional argument: 'Loader'"

How can this be error be stopped?