Remove unused jedi imports

Restore compatibility with the latest jedi.
Merge pull request #118 from vlevit/jedi-import-fix

Restore compatibility with the latest jedi