Fix CI

fed/config.py

@@ -135,8 +135,10 @@ def from_dict(cls, data: Dict):
             CrossSiloMessageConfig: An instance of CrossSiloMessageConfig.
         """
         # Get the attributes of the class
+
         data = data or {}
-        attrs = {attr for attr, _ in cls.__annotations__.items()}
+        all_annotations = {**cls.__annotations__, **cls.__base__.__annotations__}
+        attrs = {attr for attr, _ in all_annotations.items()}
         # Filter the dictionary to only include keys that are attributes of the class
         filtered_data = {key: value for key, value in data.items() if key in attrs}
         return cls(**filtered_data)

tests/test_exit_on_failure_sending.py

@@ -46,7 +46,7 @@ def get_value(self):
         return self._value
 
 
-def run(party, is_inner_party):
+def run(party):
     signal.signal(signal.SIGTERM, signal_handler)
 
     compatible_utils.init_ray(address='local')
@@ -83,7 +83,7 @@ def run(party, is_inner_party):
 
 
 def test_exit_when_failure_on_sending():
-    p_alice = multiprocessing.Process(target=run, args=('alice', True))
+    p_alice = multiprocessing.Process(target=run, args=('alice',))
     p_alice.start()
     p_alice.join()
     assert p_alice.exitcode == 0