Fix docker run

Signed-off-by: Chaurasiya, Payal <[email protected]>
payalcha · Jan 22, 2025 · 6658aa8 · 6658aa8
1 parent 61d867c
commit 6658aa8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/end_to_end/test_suites/tr_with_eval_tests.py b/tests/end_to_end/test_suites/tr_with_eval_tests.py
@@ -82,8 +82,8 @@ def test_eval_federation_via_dockerized_workspace(request, fx_federation_tr_dws)
     ), "Federation completion failed"
 
     # Set the best model path
-    request.config.best_model_path = os.path.join(fx_federation_tr.aggregator.workspace_path, "save", "best.pbuf")
-    metric_file_path = os.path.join(fx_federation_tr.aggregator.workspace_path, "logs", "aggregator_metrics.txt")
+    request.config.best_model_path = os.path.join(fx_federation_tr_dws.aggregator.workspace_path, "save", "best.pbuf")
+    metric_file_path = os.path.join(fx_federation_tr_dws.aggregator.workspace_path, "logs", "aggregator_metrics.txt")
     model_accuracy = get_aggregated_accuracy(metric_file_path)
     log.info(f"Model accuracy post {request.config.num_rounds} rounds: {model_accuracy}")