Thank you for releasing this extremely well-documented and pedagogical codebase -- I'm sure that students like myself will be learning A LOT from it over the coming years!
Hey @chaitjo Thanks for interests and PR!! These changes look good to me! However, we have not figured out a protocol to accept external PR yet. Will discuss with the team next week and get back to you!
This PR fixes some minor docstring issues.
Thank you for releasing this extremely well-documented and pedagogical codebase -- I'm sure that students like myself will be learning A LOT from it over the coming years!