Polish

gt_matlab_trajs.py

@@ -64,14 +64,13 @@ for edge in g.edges():
 
 print('Algorithm done, saving...')
 
-# make a dictionary of trajectories:
+# Make a dictionary of trajectories:
 
 trajectories = {}
 
 for node in g.vertices():
     if g.vp.traj[node] not in trajectories:
-        trajectories[g.vp.traj[node]] = []
-        trajectories[g.vp.traj[node]].append(g.vp.name[node])
+        trajectories[g.vp.traj[node]] = [g.vp.name[node]]
     else:
         trajectories[g.vp.traj[node]].append(g.vp.name[node])