Gathered all Arithmetic Rules.

linqs · Aug 22, 2020 · 731cd97 · 731cd97
1 parent 801b562
commit 731cd97
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/psl-core/src/main/java/org/linqs/psl/model/rule/arithmetic/AbstractArithmeticRule.java b/psl-core/src/main/java/org/linqs/psl/model/rule/arithmetic/AbstractArithmeticRule.java
@@ -289,6 +289,9 @@ private void groundForNonSummation(Constant[] constants, Map<Variable, Integer>
         groundSingleNonSummationRule(constants, variableMap, atomManager, resources);
 
         results.addAll(resources.groundRules);
+        for (GroundRule groundRule : resources.groundRules) {
+            VizDataCollection.addGroundRule(this, groundRule, variableMap, constants);
+        }
         resources.groundRules.clear();
         resources.accessExceptionAtoms.clear();
     }
@@ -310,6 +313,9 @@ private void groundForSummation(Constant[] constants, Map<Variable, Integer> var
         groundSingleSummationRule(constants, variableMap, atomManager, resources);
 
         results.addAll(resources.groundRules);
+        for (GroundRule groundRule : resources.groundRules) {
+            VizDataCollection.addGroundRule(this, groundRule, variableMap, constants);
+        }
         resources.groundRules.clear();
         resources.accessExceptionAtoms.clear();
     }