FIX: grounding reordering for FOL Not node

Signed-off-by: naweedkhan <[email protected]>
IBM · Nov 30, 2023 · 687b86c · 687b86c
1 parent c0f6332
commit 687b86c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lnn/symbolic/logic/unary_operator.py b/lnn/symbolic/logic/unary_operator.py
@@ -416,7 +416,8 @@ def downward(self, **kwds) -> torch.Tensor:
                 if g not in self.operands[0]._groundings:
                     self.operands[0]._add_groundings(g)
         bounds = self.operands[0].neuron.aggregate_bounds(
-            None, _utils.negate_bounds(self.get_data(*groundings))
+            [self.operands[0].grounding_table[_] for _ in groundings],
+            _utils.negate_bounds(self.get_data(*groundings))
         )
         if self.operands[0].is_contradiction():
             self.logger.info(