From 85f92cd0f6737ff8d65b2f00945a7b8fccdcf536 Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Thu, 7 Jul 2016 23:14:55 -0400 Subject: [PATCH] Check whether original graph has edge between separated fragments --- gala/serve.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/gala/serve.py b/gala/serve.py index d24a5f4..1a112b0 100644 --- 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