From 0ccf4dbf732a34dfc4f7f8d1baed332ca9da0a4d Mon Sep 17 00:00:00 2001
From: Colin <cophus@gmail.com>
Date: Sat, 24 Aug 2024 15:32:05 -0700
Subject: [PATCH] ability to return fig handles

---
 py4DSTEM/process/polar/polar_peaks.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/py4DSTEM/process/polar/polar_peaks.py b/py4DSTEM/process/polar/polar_peaks.py
index d2c45c530..64f730581 100644
--- 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,