Bug: Import statements from Pymatgen are outdated w/ newest version

PR is up. Thanks, Anubhav :slight_smile: Wasn’t sure if a random PR would be welcome without an explanation / flag beforehand.