
import networkx as nx


class StellarGraph(nx.MultiDiGraph):
    """
    Our own StellarGraph class, inherited from nx.MultiDiGraph, with extra stuff to be added that's needed by samplers and mappers
    """

    pass


import networkx as nx


class StellarGraph(nx.MultiGraph):
    """
    Our own class for heterogeneous undirected graphs, inherited from nx.MultiGraph, with extra stuff to be added that's needed by samplers and mappers
    """

    def __init__(self):
        super().__init__()


class StellarDiGraph(nx.MultiDiGraph):
    """
    Our own class for heterogeneous directed graphs, inherited from nx.MultiDiGraph, with extra stuff to be added that's needed by samplers and mappers
    """

    def __init__(self):
        super().__init__()



Split StellarGraph class into two: undirected and directed graph classes
Split StellarGraph class into two: undirected and directed graph classes

  
  import networkx as nx
  
  
- class StellarGraph(nx.MultiDiGraph):
+ class StellarGraph(nx.MultiGraph):
      """
-     Our own StellarGraph class, inherited from nx.MultiDiGraph, with extra stuff to be added that's needed by samplers and mappers
+     Our own class for heterogeneous undirected graphs, inherited from nx.MultiGraph, with extra stuff to be added that's needed by samplers and mappers
      """
  
-     pass
+     def __init__(self):
+         super().__init__()
  
+ 
+ class StellarDiGraph(nx.MultiDiGraph):
+     """
+     Our own class for heterogeneous directed graphs, inherited from nx.MultiDiGraph, with extra stuff to be added that's needed by samplers and mappers
+     """
+ 
+     def __init__(self):
+         super().__init__()
+ 
+ 
+ 
Split StellarGraph class into two: undirected and directed graph classes
## Code Before:

import networkx as nx


class StellarGraph(nx.MultiDiGraph):
    """
    Our own StellarGraph class, inherited from nx.MultiDiGraph, with extra stuff to be added that's needed by samplers and mappers
    """

    pass

## Instruction:
Split StellarGraph class into two: undirected and directed graph classes
## Code After:

import networkx as nx


class StellarGraph(nx.MultiGraph):
    """
    Our own class for heterogeneous undirected graphs, inherited from nx.MultiGraph, with extra stuff to be added that's needed by samplers and mappers
    """

    def __init__(self):
        super().__init__()


class StellarDiGraph(nx.MultiDiGraph):
    """
    Our own class for heterogeneous directed graphs, inherited from nx.MultiDiGraph, with extra stuff to be added that's needed by samplers and mappers
    """

    def __init__(self):
        super().__init__()


