Update viz_scrna.py

Remove custom ticklabels on density video. Fix Issue #7.
KrishnaswamyLab · Oct 15, 2020 · ca8ac7b · ca8ac7b
1 parent 21d11e3
commit ca8ac7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/TrajectoryNet/lib/viz_scrna.py b/TrajectoryNet/lib/viz_scrna.py
@@ -352,10 +352,10 @@ def save_trajectory_density(prior_logdensity, model, data_samples, savedir, ntim
                         cmap=plt.cm.Spectral,
                         orientation='horizontal')
                 cb.set_ticks(np.linspace(0,1,4))
-                cb.set_ticklabels(['48HR', 'Day 12', 'Day 18', 'Day 30'])
+                #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(5))
+                cb.set_ticklabels(np.arange(len(end_times))
                 ax.axvline(t / grid_z_traj.shape[0], c='k', linewidth=15)
                 ax.set_title('Time')