diff --git a/fedn/fedn/network/clients/client.py b/fedn/fedn/network/clients/client.py index 7b19fd88b..633b2078a 100644 --- a/fedn/fedn/network/clients/client.py +++ b/fedn/fedn/network/clients/client.py @@ -568,11 +568,6 @@ def _process_training_request(self, model_id): updated_model_id = None meta = {'status': 'failed', 'error': str(e)} - # Push model update to combiner server - updated_model_id = uuid.uuid4() - self.set_model(out_model, str(updated_model_id)) - meta['upload_model'] = time.time() - tic - self.state = ClientState.idle return updated_model_id, meta diff --git a/fedn/fedn/network/combiner/roundhandler.py b/fedn/fedn/network/combiner/roundhandler.py index aa944a058..a3acaf20e 100644 --- a/fedn/fedn/network/combiner/roundhandler.py +++ b/fedn/fedn/network/combiner/roundhandler.py @@ -343,6 +343,7 @@ def execute_training_round(self, config): logger.info( "ROUNDCONTROL: TRAINING ROUND COMPLETED. Aggregated model id: {}, Job id: {}".format(model_id, config['_job_id'])) + self.modelservice.models.delete(config['model_id']) return data def run(self, polling_interval=1.0):