Output figure object in visualize_field

MassimoCimmino · Dec 5, 2017 · 2523f67 · 2523f67
1 parent d4f4be8
commit 2523f67
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pygfunction/boreholes.py b/pygfunction/boreholes.py
@@ -408,6 +408,11 @@ def visualize_field(borefield):
     borefield : list
         List of boreholes in the bore field.
 
+    Returns
+    -------
+    fig : figure
+        Figure object (matplotlib).
+
     """
     import matplotlib.pyplot as plt
     from matplotlib.ticker import AutoMinorLocator
@@ -473,6 +478,6 @@ def visualize_field(borefield):
     ax1.yaxis.set_minor_locator(AutoMinorLocator())
     ax1.zaxis.set_minor_locator(AutoMinorLocator())
 
-    plt.tight_layout()
+    plt.tight_layout(rect=[0, 0.0, 0.95, 1.0])
 
-    return borefield
+    return fig