Добавьте подсказки типов при инициализации django

У меня есть пользователь в качестве внешнего ключа user = models.ForeignKey(settings.AUTH_USER_MODEL, models.CASCADE) Но когда я обращаюсь к пользователю через связанный объект, я не вижу автозаполнения для атрибутов пользователя. Например Unresolved attribute reference 'public_name' for class 'ForeignKey' Как я могу добавить подсказку о том, что это модель User?
В инициализации приложения django я переопределяю serializers.ModelSerializer с моим пользовательским сериализатором. И переопределение происходит в модуле app/
init.py. Проблема в том, что я не могу увидеть некоторые атрибуты, которые существуют в CustomModelSerializer. Pycharm думает, что это ModelSerializer из drf. Можно ли где-то намекнуть, что это CustomModelSerializer. (PS: Я знаю, что могу импортировать CustomModelSerializer куда угодно, но это большой проект и я искал решения с добавлением подсказки типа или заставить Pycharm распознать переопределение) .