Fix datagen timeout problems

Some trajectories are timing out well beyond the specified config time limit per trajectory, this makes it difficult to predict what the run time will be on zaratan, this ticket will investigate this issue.