In case you’ve followed the current version of the README you should install our code with pip install -e . instead of pip install . as it currently says. I’ve been meaning to open a PR to fix that You need the -e to get it to reflect any changes you make.
minor change via @jayqi