Which version of PyYaml needed/how to install?

I'm getting this error when following instructions in the GitHub repo you provided to run your project via `poetry run python run.py`

Traceback (most recent call last):
File "/home/ighodgao/determinedMalwareBuster/run.py", line 41, in
context = yaml.load(open("experiments/const.yml", "r"))
TypeError: load() missing 1 required positional argument: 'Loader'

PyYaml was missing in the .toml file, so I added in version 6.0 which is the latest version and then re-ran with the updated poetry installation. Can you let me know which version of PyYaml (or other yaml handler) you may have used ASAP?

Thanks!
Isha