Fix remote calling of load_model() (#136)

mc2-project · Mar 17, 2021 · aa7093d · aa7093d
1 parent 82f2d1a
commit aa7093d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/demo/python/remote-control/client/client.py b/demo/python/remote-control/client/client.py
@@ -55,6 +55,9 @@ def run(channel_addr, sym_key_file, priv_key_file, cert_file):
 
     booster.save_model(HOME_DIR + "demo/python/remote-control/client/modelfile.model")
 
+    booster = xgb.Booster()
+    booster.load_model(HOME_DIR + "demo/python/remote-control/client/modelfile.model")
+
     # Get encrypted predictions
     print("\nModel Predictions: ")
     predictions, num_preds = booster.predict(dtest, decrypt=False)

diff --git a/python-package/securexgboost/core.py b/python-package/securexgboost/core.py
@@ -3007,7 +3007,7 @@ def XGBoosterLoadModel(request, signers, signatures, sig_lengths):
             ctypes.c_uint32(nonce_ctr),
             ctypes.byref(out_sig),
             ctypes.byref(out_sig_len),
-            signers,
+            from_pystr_to_cstr(signers),
             c_signatures,
             c_sig_lengths))
         return out_sig, out_sig_len.value