Browse Source

Merge pull request #173 from openfheorg/166-fix-print-statement

166 fixed a print statement and added more functions to binding
Muthu Annamalai (முத்து அண்ணாமலை) 1 year ago
parent
commit
79d5dac7ac
2 changed files with 4 additions and 2 deletions
  1. 1 1
      examples/pke/simple-ckks-bootstrapping.py
  2. 3 1
      src/lib/bindings.cpp

+ 1 - 1
examples/pke/simple-ckks-bootstrapping.py

@@ -66,7 +66,7 @@ def simple_bootstrap_example():
 
     ciphertext_after = cryptocontext.EvalBootstrap(ciph)
 
-    print(f"Number of levels remaining after bootstrapping: {depth - ciphertext_after.GetLevel()}")
+    print(f"Number of levels remaining after bootstrapping: {depth - ciphertext_after.GetLevel() - (ciphertext_after.GetNoiseScaleDeg() - 1)}")
 
     result = cryptocontext.Decrypt(ciphertext_after,key_pair.secretKey)
     result.SetLength(encoded_length)

+ 3 - 1
src/lib/bindings.cpp

@@ -1114,7 +1114,9 @@ void bind_ciphertext(py::module &m)
     // .def("GetScalingFactor", &CiphertextImpl<DCRTPoly>::GetScalingFactor)
     // .def("SetScalingFactor", &CiphertextImpl<DCRTPoly>::SetScalingFactor)
      .def("GetSlots", &CiphertextImpl<DCRTPoly>::GetSlots)
-     .def("SetSlots", &CiphertextImpl<DCRTPoly>::SetSlots);
+     .def("SetSlots", &CiphertextImpl<DCRTPoly>::SetSlots)
+     .def("GetNoiseScaleDeg", &CiphertextImpl<DCRTPoly>::GetNoiseScaleDeg)
+     .def("SetNoiseScaleDeg", &CiphertextImpl<DCRTPoly>::SetNoiseScaleDeg);
 }
 
 void bind_schemes(py::module &m){