pryvx 2.4.6__py3-none-any.whl → 2.4.8__py3-none-any.whl

pryvx/fl_client.py

@@ -14,9 +14,11 @@ def train_logistic_classifier(features, labels):
 
 
 def train_random_forest_classifier(features, labels):
+    print("Model training started...")
     model = RandomForestClassifier(n_estimators=100, random_state=42)
     model.fit(features, labels)
     serialized_model = pickle.dumps(model)
+    print("Model training finished.")
     return serialized_model, model
 
 

pryvx/fl_server.py

@@ -4,11 +4,18 @@ from pryvx import pryvx_pb2_grpc
 import pickle
 from concurrent import futures
 import os
-import datetime
+import shutil
 
 # Directory to store received models
 MODEL_SAVE_PATH = "received_models/"
-os.makedirs(MODEL_SAVE_PATH, exist_ok=True)
+
+def prepare_model_folder():
+    # Delete the folder if it exists
+    if os.path.exists(MODEL_SAVE_PATH):
+        shutil.rmtree(MODEL_SAVE_PATH)
+    # Create a new folder
+    os.makedirs(MODEL_SAVE_PATH, exist_ok=True)
+
 
 class ModelServicer(pryvx_pb2_grpc.ModelServiceServicer):
     def __init__(self):
@@ -41,8 +48,11 @@ class ModelServicer(pryvx_pb2_grpc.ModelServiceServicer):
 
 
 def start_server():
+    prepare_model_folder()
+
     server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
     pryvx_pb2_grpc.add_ModelServiceServicer_to_server(ModelServicer(), server)
+    
     server.add_insecure_port('[::]:50051')
     server.start()
     print("Server started. Listening on localhost:50051")

{pryvx-2.4.6.dist-info → pryvx-2.4.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pryvx
-Version: 2.4.6
+Version: 2.4.8
 Summary: A comprehensive package for privacy-enhancing technologies
 Home-page: UNKNOWN
 Author: PryvX (Jayesh Kenaudekar)

{pryvx-2.4.6.dist-info → pryvx-2.4.8.dist-info}/RECORD

@@ -1,7 +1,7 @@
 pryvx/__init__.py,sha256=QvpYISJ8MH7XnbxC5iTMTycq7Svsld0NIbMozcndYjo,111
-pryvx/fl_client.py,sha256=gQNjv99tDDjaGj0JTdAVomefqR2F3_xNLcnHJ6aGZZo,1152
+pryvx/fl_client.py,sha256=ksJM2R5adi8P7No1t4mhpMiwr-ypmsgvNL14LmfqOyg,1231
 pryvx/fl_ensemble.py,sha256=lvgvd-zJDSBh-RyMAjTyJZwu0eFiUNXfCU7J2Vk7g0g,715
-pryvx/fl_server.py,sha256=JqYtlLtLTcKxrNB8oY-hdZmG8VajS7mIqAY3OjOUxmU,1822
+pryvx/fl_server.py,sha256=LJdJkj2BJdXrMczy1qZBHN_fY-ZveH-vHvljwwgDv8k,2039
 pryvx/gdp.py,sha256=ToLtWNvBKtSs-3Na1pGe_J6dWvBxXhV2raUJkNtKMg4,925
 pryvx/ldp.py,sha256=WxBQLnZ_b2wNV1rlN7rtKXaQ5waUPPvnRus-4tUnCB8,810
 pryvx/phe.py,sha256=HN70aUNTq0w9vRPMaNGfTnNi00gMQ1B7R7Cm6r6rheE,2974
@@ -9,8 +9,8 @@ pryvx/pryvx_pb2.py,sha256=HlwY1zRVFReIaK8WZ8opJ1kd3_PQWSJEx-PZImWYlCM,1621
 pryvx/pryvx_pb2_grpc.py,sha256=yn4mThL3XY8glPvZ9b1Fmj_c8JtAs1pTNp3m0JV9d3s,3425
 pryvx/psi.py,sha256=V0BUJwYfiFWhKtEL27q45yxY7U-I7B4olKQhaGbgN50,1346
 pryvx/smpc.py,sha256=FZcE3gZo8WUVPmcBHSgP8iOkWt3AKcBLSLCtKczy8hs,2247
-pryvx-2.4.6.dist-info/LICENSE,sha256=pbIMXbaorAIVW-fDch2tvtZRkVA3mz-UnXQqeCf4LDg,1086
-pryvx-2.4.6.dist-info/METADATA,sha256=zXQ9VgjJkkJRHsBoy5b1lfEf4y5CHjwzxqhTxD0iBlw,1066
-pryvx-2.4.6.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-pryvx-2.4.6.dist-info/top_level.txt,sha256=1iHoeevu_FoFjdPg8HyMlgvHNUm1--9QEpFMaJEa4hw,6
-pryvx-2.4.6.dist-info/RECORD,,
+pryvx-2.4.8.dist-info/LICENSE,sha256=pbIMXbaorAIVW-fDch2tvtZRkVA3mz-UnXQqeCf4LDg,1086
+pryvx-2.4.8.dist-info/METADATA,sha256=JNelEJzGkMg8HxZREkwli6wd_lVmJhP58SHmW5Wa0LE,1066
+pryvx-2.4.8.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+pryvx-2.4.8.dist-info/top_level.txt,sha256=1iHoeevu_FoFjdPg8HyMlgvHNUm1--9QEpFMaJEa4hw,6
+pryvx-2.4.8.dist-info/RECORD,,