Avoid loading the dataset in the parent process when running the simulator within the same framework
When running the simulator through the bash scripts, each step in the pipeline is executed in its own Python process. However, when all steps are executed within the same process, the training dataset is loaded in the parent process in the download()
method.
A simple solution could be to push the download of the dataset to a subprocess as well.