Update database.py

eguidotti · Feb 11, 2024 · 76fe2f1 · 76fe2f1
1 parent faad5c3
commit 76fe2f1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/bornrule/sql/database.py b/bornrule/sql/database.py
@@ -383,8 +383,6 @@ def _sql_transform(self, items, concat, name):
                 1.0 AS {self.w}
             FROM 
                 {table}
-            WHERE
-                {items[2]} IS NOT NULL
             """
 
     def _sql_partial_fit(self, items, sample_weight):
@@ -406,7 +404,8 @@ def _sql_partial_fit(self, items, sample_weight):
             FROM
                 P_jk
             WHERE
-                1 = 1
+                {self.j} IS NOT NULL AND
+                {self.k} IS NOT NULL
             """
 
     def _sql_predict(self, items, cache):