fixed weird embed tool bug

franklee26 · Jun 17, 2019 · 7162070 · 7162070
1 parent c88b6c3
commit 7162070
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/EMBED_TOOLS.py b/EMBED_TOOLS.py
@@ -17,7 +17,7 @@ def DELETE_NODES(self, nodes):
 		TEMP.add_edges_from(self.SHALLOW_COPY.edges())
 		# usable version
 		for n in nodes:
-			toRestore = [nn for list(TEMP.neighbors(n)) in n1 if len(list(TEMP.neighbors(nn))) == 1]
+			toRestore = [nn for nn in list(TEMP.neighbors(n)) if len(list(TEMP.neighbors(nn))) == 1]
 			TEMP.remove_node(n)
 			for nn in toRestore:
 				TEMP.add_node(nn)

diff --git a/Embed.py b/Embed.py
@@ -4,7 +4,8 @@
 import math
 import random
 # for general tools: Python passes objects by reference!
-import QEmbed.EMBED_TOOLS as ET
+#import QEmbed.EMBED_TOOLS as ET
+import EMBED_TOOLS as ET
 
 # Custom exceptions