Fix missing import in parallel_parameter_sweep.py