a plausibly working version of this script on my machine. had to uninstall conda's version of yt.