Update GeneralGraph.py

made a mistake in variable assignment
updating subgraph to not have axis out of bound error

Signed-off-by: JanMarcoRuizdeVargas <[email protected]>

Author: JanMarcoRuizdeVargas

causallearn/graph/GeneralGraph.py

@@ -874,8 +874,8 @@ def subgraph(self, nodes: List[Node]):
             if not (self.nodes[i] in nodes):
                 nodes_to_delete .append(i)
 
-        np.delete(graph, nodes_to_delete, axis = 0)
-        np.delete(graph, nodes_to_delete, axis = 1)
+        graph = np.delete(graph, nodes_to_delete, axis = 0)
+        graph = np.delete(graph, nodes_to_delete, axis = 1)
 
         subgraph.graph = graph
         subgraph.reconstitute_dpath(subgraph.get_graph_edges())