Strange looking type of trajs in data_generation.py

After the fix in data_generation.py we get a strange looking type of trajs and meas_trajs Just want to make sure if that is fine.