fixing grpc

scaleoutsystems · Dec 20, 2024 · 78096ad · 78096ad
1 parent 81eae71
commit 78096ad
Show file tree

Hide file tree

Showing 3 changed files with 103 additions and 102 deletions.
diff --git a/fedn/network/clients/fedn_client.py b/fedn/network/clients/fedn_client.py
@@ -362,7 +362,7 @@ def forward_embeddings(self, request):
 
         self.send_status(
             "Forward pass completed.",
-            log_level=fedn.Status.AUDIT,
+            log_level=fedn.LogLevel.AUDIT,
             type=fedn.StatusType.MODEL_UPDATE,
             request=update,
             sesssion_id=request.session_id,
@@ -407,7 +407,7 @@ def backward_gradients(self, request):
 
             self.send_status(
                 "Backward pass completed.",
-                log_level=fedn.Status.AUDIT,
+                log_level=fedn.LogLevel.AUDIT,
                 type=fedn.StatusType.BACKWARD,
                 # request=update,
                 sesssion_id=request.session_id,

diff --git a/fedn/network/grpc/fedn.proto b/fedn/network/grpc/fedn.proto
@@ -16,10 +16,11 @@ enum StatusType {
   MODEL_VALIDATION_REQUEST = 3;
   MODEL_VALIDATION = 4;
   MODEL_PREDICTION = 5;
-  FORWARD_REQUEST = 6;
-  FORWARD = 7;
-  BACKWARD_REQUEST = 8;
-  BACKWARD = 9;
+  NETWORK = 6;
+  FORWARD_REQUEST = 7;
+  FORWARD = 8;
+  BACKWARD_REQUEST = 9;
+  BACKWARD = 10;
 }
 
 enum LogLevel {