Bandit issue resolved

Signed-off-by: noopur <[email protected]>
securefederatedai · Jan 20, 2025 · 2f31d3b · 2f31d3b
1 parent b5064eb
commit 2f31d3b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/end_to_end/utils/federation_helper.py b/tests/end_to_end/utils/federation_helper.py
@@ -7,6 +7,7 @@
 import os
 import json
 import re
+import subprocess
 import papermill as pm
 from pathlib import Path
 import shutil
@@ -710,9 +711,7 @@ def download_data(collaborators, model_name, local_bind_path):
 
     log.info("Downloading the data for the model. This will take some time to complete based on the data size ..")
     try:
-        # Data setup file has logic to show progress bar
-        # Using os.system to make sure it is displayed
-        os.system(f"cd {local_bind_path}; python {constants.DATA_SETUP_FILE} {len(collaborators)}")
+        subprocess.run(f"cd {local_bind_path}; python {constants.DATA_SETUP_FILE} {len(collaborators)}", shell=True)
     except Exception:
         raise ex.DataSetupException(f"Failed to download data for {model_name}")