Merge pull request #12 from Desilo/11-fix-cov

Removed rescale_every from cov
Desilo · Jan 3, 2024 · 82567d6 · 82567d6
2 parents 0d71574 + 9cb001c
commit 82567d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/liberate/fhe/ckks_engine.py b/src/liberate/fhe/ckks_engine.py
@@ -2316,9 +2316,9 @@ def mean(self, ct, gk, alpha=1):
         return new_ct
 
     def cov(self, ct_a: data_struct, ct_b: data_struct,
-            evk: data_struct, gk: data_struct, rescale_every=5) -> data_struct:
-        cta_mean = self.mean(ct_a, gk, rescale_every=rescale_every)
-        ctb_mean = self.mean(ct_b, gk, rescale_every=rescale_every)
+            evk: data_struct, gk: data_struct) -> data_struct:
+        cta_mean = self.mean(ct_a, gk)
+        ctb_mean = self.mean(ct_b, gk)
 
         cta_dev = self.sub(ct_a, cta_mean)
         ctb_dev = self.sub(ct_b, ctb_mean)