fix

scaleoutsystems · Jan 29, 2025 · aab9db1 · aab9db1
1 parent 34d82c3
commit aab9db1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fedn/network/clients/fedn_client.py b/fedn/network/clients/fedn_client.py
@@ -7,7 +7,7 @@
 import time
 import uuid
 from io import BytesIO
-from typing import Any, Optional, Tuple
+from typing import Any, Optional, Tuple, Union
 
 import requests
 
@@ -426,8 +426,8 @@ def send_status(
         self,
         msg: str,
         log_level: fedn.LogLevel = fedn.LogLevel.INFO,
-        type: Optional[fedn.StatusType] = None,
-        request: Optional[fedn.TaskRequest] = None,
+        type: Optional[str] = None,
+        request: Optional[Union[fedn.ModelUpdate, fedn.ModelValidation, fedn.TaskRequest]] = None,
         sesssion_id: Optional[str] = None,
         sender_name: Optional[str] = None,
     ) -> None: