Skip to content

Commit

Permalink
Log correction
Browse files Browse the repository at this point in the history
Signed-off-by: noopur <[email protected]>
  • Loading branch information
noopurintel committed Nov 20, 2024
1 parent 5ce09b3 commit cabdbf7
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions tests/end_to_end/utils/federation_helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def setup_pki(fed_obj):
fed_obj.aggregator.generate_sign_request()
fed_obj.model_owner.certify_aggregator(fed_obj.aggregator.agg_domain_name)
except Exception as e:
log.error(f"Failed to perform aggregator operations: {e}")
log.error(f"Failed to perform PKI setup for aggregator: {e}")
raise e

# Collaborator and model owner operations
Expand All @@ -38,11 +38,11 @@ def setup_pki(fed_obj):
fed_obj.model_owner.certify_collaborator(collaborator.collaborator_name)
collaborator.import_pki()
except Exception as e:
log.error(f"Failed to perform collaborator operations: {e}")
log.error(f"Failed to perform PKI setup for {collaborator.collaborator_name}: {e}")
raise e
success = True

log.info("CSR operations completed successfully for all participants")
log.info("PKI setup successfully for all participants")
return success


Expand Down Expand Up @@ -93,8 +93,7 @@ def verify_federation_run_completion(fed_obj, results):
# Result will contain a list of boolean values for all the participants.
# True - successful completion, False - failed/incomplete
results = [f.result() for f in futures]
log.info(f"Results: {results}")

log.info(f"Results from all the participants: {results}")
# If any of the participant failed, raise exception
if not all(results):
log.error("Federation run completion failed/timed out for one or more participants")
Expand Down

0 comments on commit cabdbf7

Please sign in to comment.