[FIX] Code style in models' Python files