diff --git a/api/internal_services/background_worker.py b/api/internal_services/background_worker.py index 5ce476ab7cb2ffb91735bb0539b199d3289ab54a..8ed9d8adde2a34698cc6c0c088fee9523d235c91 100644 --- a/api/internal_services/background_worker.py +++ b/api/internal_services/background_worker.py @@ -53,6 +53,7 @@ def process_trainer(): finetuned_repo_name=job.job_data['finetuned_repo_name'], huggingface_token=job.job_data['huggingface_token'], ) + logger.debug(request) responses = stub.StartTraining(request) for response in responses: logger.debug(f"gRPC message : {response.status}") diff --git a/microservices/trainer/trainer.py b/microservices/trainer/trainer.py index 8016b71ff200dfcb233d565c1a94de47079939ed..7e7919e5949c3f01b46074ed97366668a8a1914f 100644 --- a/microservices/trainer/trainer.py +++ b/microservices/trainer/trainer.py @@ -28,11 +28,12 @@ class TrainerServicer(trainer_pb2_grpc.TrainerServicer): def StartTraining(self, request, context): def response_messages(): - print("reception d'un event") + print("event received") global is_busy if not is_busy: is_busy = True + print(f"incoming request : {request}") training_process( training_data=request.training_data, fondation_model_id=request.fondation_model_id,