Update models.py

Import oscar partner models