added fig.show() in plotFluxPathways. This fixes #246

Acellera · Feb 2, 2017 · dd91d41 · dd91d41
1 parent fab84a3
commit dd91d41
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/htmd/kinetics.py b/htmd/kinetics.py
@@ -295,7 +295,7 @@ def plotFluxPathways(self, statetype='macro', mode='net_flux', fraction=1.0):
 
         if statetype == 'micro':
             tpt = msm.tpt(self.model.msm, [self.sourcemicro], [self.sinkmicro])
-            plot_flux(tpt, attribute_to_plot=mode)
+            fig, pos = plot_flux(tpt, attribute_to_plot=mode)
         elif statetype == 'macro' or statetype == 'coarse':
             metastable_sets = []
             for i in range(self.model.macronum):
@@ -312,7 +312,8 @@ def plotFluxPathways(self, statetype='macro', mode='net_flux', fraction=1.0):
                         setmap.append(idx2)
                         continue
             setmap = np.array(setmap)
-            plot_flux(tpt, attribute_to_plot=mode, state_labels=setmap)
+            fig, pos = plot_flux(tpt, attribute_to_plot=mode, state_labels=setmap)
+        fig.show()
 
         paths, pathfluxes = tpt.pathways(fraction=fraction)
         cumflux = 0