Check whether original graph has edge between separated fragments

jni · Jul 8, 2016 · 85f92cd · 85f92cd
1 parent 26ae97c
commit 85f92cd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/gala/serve.py b/gala/serve.py
@@ -224,9 +224,10 @@ def learn_separation(self, fragment, separate_from):
             if self.rag.has_edge(s0, s1):
                 self.features.append(self.feature_manager(self.rag, s0, s1))
                 self.targets.append(SEPAR_LABEL)
-            self.features.append(self.feature_manager(self.original_rag,
-                                                      f0, f1))
-            self.targets.append(SEPAR_LABEL)
+            if self.original_rag.has_edge(f0, f1):
+                self.features.append(self.feature_manager(self.original_rag,
+                                                          f0, f1))
+                self.targets.append(SEPAR_LABEL)
             self.separate.append((f0, f1))
         self.recently_solved = False