BUG Get axis instead of figure.

quantopian · Apr 15, 2019 · 712716a · 712716a
1 parent c939ceb
commit 712716a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyfolio/plotting.py b/pyfolio/plotting.py
@@ -1213,7 +1213,7 @@ def plot_max_median_position_concentration(positions, ax=None, **kwargs):
     """
 
     if ax is None:
-        ax = plt.gcf()
+        ax = plt.gca()
 
     alloc_summary = pos.get_max_median_position_concentration(positions)
     colors = ['mediumblue', 'steelblue', 'tomato', 'firebrick']
@@ -1249,7 +1249,7 @@ def plot_sector_allocations(returns, sector_alloc, ax=None, **kwargs):
     """
 
     if ax is None:
-        ax = plt.gcf()
+        ax = plt.gca()
 
     sector_alloc.plot(title='Sector allocation over time',
                       alpha=0.5, ax=ax, **kwargs)