diff --git a/nvflare/private/fed/app/simulator/simulator_runner.py b/nvflare/private/fed/app/simulator/simulator_runner.py index 0976ade155..fc4a2edd7f 100644 --- a/nvflare/private/fed/app/simulator/simulator_runner.py +++ b/nvflare/private/fed/app/simulator/simulator_runner.py @@ -709,9 +709,10 @@ def do_one_task(self, client, num_of_threads, gpu, lock, timeout=60.0, task_name command += " --gpu " + str(gpu) new_env = os.environ.copy() add_custom_dir_to_path(app_custom_folder, new_env) - if self.server_custom_folder: + if os.path.isdir(self.server_custom_folder): python_paths = new_env[SystemVarName.PYTHONPATH].split(os.pathsep) - python_paths.remove(self.server_custom_folder) + if self.server_custom_folder in python_paths: + python_paths.remove(self.server_custom_folder) new_env[SystemVarName.PYTHONPATH] = os.pathsep.join(python_paths) _ = subprocess.Popen(shlex.split(command, True), preexec_fn=os.setsid, env=new_env)