Remove unused import from diagnose.py (Fixes #266)