Add missing import in graphs models.py