ability to return fig handles

py4dstem · Aug 24, 2024 · 0ccf4db · 0ccf4db
1 parent 3e58443
commit 0ccf4db
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/py4DSTEM/process/polar/polar_peaks.py b/py4DSTEM/process/polar/polar_peaks.py
@@ -1104,6 +1104,7 @@ def plot_crystal_amorphous_fraction(
     legend=True,
     ticks=False,
     figsize=(5, 4),
+    returnfig=False,
 ):
     """
     Plotting function for the crystal / amorphous fraction image.
@@ -1157,6 +1158,8 @@ def plot_crystal_amorphous_fraction(
         )
         cbar.ax.set_ylabel("More Amorphous  <---->  More Crystalline")
 
+    if returnfig:
+        return fig,ax
 
 def refine_peaks(
     self,