Мне нужно расширить пакет Python Networkx и добавить несколько методовGraph класс для моей особой потребности

То, как я думал об этом, просто вывод нового классаNewGraphи добавив необходимые методы.

Однако в networkx есть несколько других функций, которые создают и возвращаютGraph объекты (например, генерировать случайный граф). Теперь мне нужно повернуть этиGraph объекты вNewGraph объекты, чтобы я мог использовать свои новые методы.

Каков наилучший способ сделать это? Или я должен решать проблему совершенно по-другому?