Update viz_scrna.py

missing close paren whoops! #10

Author: atong01

TrajectoryNet/lib/viz_scrna.py

@@ -355,7 +355,7 @@ def save_trajectory_density(prior_logdensity, model, data_samples, savedir, ntim
                 #cb.set_ticklabels(['48HR', 'Day 12', 'Day 18', 'Day 30'])
                 #cb.set_ticklabels(['E12.5', 'E14.5', 'E16.0', 'E17.5'])
                 #cb.set_ticks(np.linspace(0,1,5))
-                cb.set_ticklabels(np.arange(len(end_times))
+                cb.set_ticklabels(np.arange(len(end_times)))
                 ax.axvline(t / grid_z_traj.shape[0], c='k', linewidth=15)
                 ax.set_title('Time')