Logging fix

Signed-off-by: noopur <[email protected]>

.github/workflows/task_runner_e2e.yml

@@ -79,7 +79,7 @@ jobs:
       id: run_task_runner_tests
       run: |
         pytest -v tests/openfl_e2e/test_suites/task_runner_tests.py -m ${{ env.MODEL_NAME }} -s --num_rounds $NUM_ROUNDS --num_collaborators $NUM_COLLABORATORS --model_name ${{ env.MODEL_NAME }}
-        echo "Task runner regression test run completed"
+        echo "Task runner e2e test run completed"
       env:
         NO_PROXY: localhost,127.0.0.1,aggregator
 

tests/openfl_e2e/utils/xml_helper.py

@@ -5,6 +5,8 @@
 from lxml import etree
 import os
 
+from tests.openfl_e2e.utils.logger import logger as log
+
 # Initialize the XML parser
 parser = etree.XMLParser(recover=True, encoding='utf-8')
 tree = ET.parse("results/results.xml", parser=parser)
@@ -61,14 +63,15 @@ def get_testcase_result():
                 database_list.append(database_dict)
                 status = None
 
-    print(f"Database list = {database_list}")
+    log.info(f"Database list = {database_list}")
     return database_list
 
 
 result = get_testcase_result()
 
 # Write the results to GitHub step summary
 with open(os.getenv('GITHUB_STEP_SUMMARY'), 'a') as fh:
+    # DO NOT change the print statements
     print("| Name | Time (in seconds) | Result |", file=fh)
     print("| ------------- | ------------- | ------------- |", file=fh)
     for item in result: