From 7a153fde0330f4c7fb9ad3ec12b1e07e7f67c84f Mon Sep 17 00:00:00 2001 From: noopur Date: Wed, 18 Dec 2024 09:21:48 +0000 Subject: [PATCH] Corrected run federation handling in memory test Signed-off-by: noopur --- tests/end_to_end/test_suites/memory_logs_tests.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/end_to_end/test_suites/memory_logs_tests.py b/tests/end_to_end/test_suites/memory_logs_tests.py index 2bb41029ab..27bde81b93 100644 --- a/tests/end_to_end/test_suites/memory_logs_tests.py +++ b/tests/end_to_end/test_suites/memory_logs_tests.py @@ -55,7 +55,12 @@ def _log_memory_usage(request, fed_obj): 6. Log the availability of memory usage details for all participants. """ # Start the federation - results = fed_helper.run_federation(fed_obj) + if request.config.test_env == "task_runner_basic": + results = fed_helper.run_federation(fed_obj) + else: + results = fed_helper.run_federation_for_dws( + fed_obj, use_tls=request.config.use_tls + ) # Verify the completion of the federation run assert fed_helper.verify_federation_run_completion(