Импортированный метод возвращает переопределенный тип

Question

Я пытаюсь визуализировать построенное дерево, используя JGraphT и JGraphX. Для этого я нашел в другом посте, что мне нужно переопределить метод toString () класса DefaultWeightedEdge

public static class MyEdge extends DefaultWeightedEdge {
        @Override
        public String toString() {
            return String.valueOf(getWeight());
        }
    }

Но теперь в моем алгоритме я должен использовать класс GusfieldGomoryHuCutTree и метод getGomoryHuTree() из библиотеки JGraphT. Этот метод возвращает только тип SimpleWeightedGraph<String, DefaultWeightedEdge>. Однако мне нужен график с переопределенным типом набора ребер, SimpleWeightedGraph<String, MyEdge>. Есть ли способ управлять этим?

Я уже пробовал обходной путь, извлекая набор ребер, вручную вводя их в MyEdge и добавляя их обратно в новый граф с теми же вершинами, но это не совсем работает, так как мне понадобится запуск / конец вершины, чтобы добавить их обратно, но методы getSource(), getTarget() из DefaultWeightedEdge оба защищены.