remove duplicates and comments

Author: AlejandroAvilesSerrano

nebula/config/config.py

@@ -77,27 +77,6 @@ def shutdown_logging(self):
 
         logging.shutdown()
 
-    def shutdown_logging(self):
-        """
-        Properly shuts down all loggers and their handlers in the system.
-        This ensures all buffered logs are written to their respective files.
-        """
-        for handler in logging.getLogger().handlers:
-            handler.flush()
-            handler.close()
-
-        training_logger = logging.getLogger(TRAINING_LOGGER)
-        for handler in training_logger.handlers:
-            handler.flush()
-            handler.close()
-
-        pl_logger = logging.getLogger("lightning.pytorch")
-        for handler in pl_logger.handlers:
-            handler.flush()
-            handler.close()
-
-        logging.shutdown()
-
     def __default_config(self):
         self.participant["device_args"]["name"] = (
             f"participant_{self.participant['device_args']['idx']}_{self.participant['network_args']['ip']}_{self.participant['network_args']['port']}"

nebula/core/engine.py

@@ -698,7 +698,6 @@ async def _start_learning(self):
                     mpe = ModelPropagationEvent(await self.cm.get_addrs_current_connections(only_direct=True, myself=False), "initialization")
                     await EventManager.get_instance().publish_node_event(mpe)
 
-                    #await self.cm.propagator.propagate("initialization")
                     await self.get_federation_ready_lock().release_async()
 
                 self.trainer.set_epochs(epochs)

nebula/core/node.py

@@ -176,20 +176,6 @@ async def main(config: Config):
     else:
         raise ValueError(f"Trainer {trainer_str} not supported")
 
-    # if config.participant["device_args"]["malicious"]:
-    #     node_cls = MaliciousNode
-    # else:
-    #     if config.participant["device_args"]["role"] == Role.AGGREGATOR.value:
-    #         node_cls = AggregatorNode
-    #     elif config.participant["device_args"]["role"] == Role.TRAINER.value:
-    #         node_cls = TrainerNode
-    #     elif config.participant["device_args"]["role"] == Role.SERVER.value:
-    #         node_cls = ServerNode
-    #     elif config.participant["device_args"]["role"] == Role.IDLE.value:
-    #         node_cls = IdleNode
-    #     else:
-    #         raise ValueError(f"Role {config.participant['device_args']['role']} not supported")
-
     VARIABILITY = 0.5
 
     def randomize_value(value, variability):