Fix style sheet examples

naoyak · naoyak · commit cfb368ab8a19 · 2017-01-20T20:38:30.000-08:00
diff --git a/examples/style_sheets/plot_bmh.py b/examples/style_sheets/plot_bmh.py
@@ -18,7 +18,6 @@
 def plot_beta_hist(ax, a, b):
     ax.hist(beta(a, b, size=10000), histtype="stepfilled",
             bins=25, alpha=0.8, normed=True)
-    return ax
 
 
 fig, ax = plt.subplots()
diff --git a/examples/style_sheets/plot_fivethirtyeight.py b/examples/style_sheets/plot_fivethirtyeight.py
@@ -11,21 +11,22 @@
 from matplotlib import pyplot as plt
 import numpy as np
 
+plt.style.use('fivethirtyeight')
+
 x = np.linspace(0, 10)
 
 # Fixing random state for reproducibility
 np.random.seed(19680801)
 
 fig, ax = plt.subplots()
 
-with plt.style.context('fivethirtyeight'):
-    ax.plot(x, np.sin(x) + x + np.random.randn(50))
-    ax.plot(x, np.sin(x) + 0.5 * x + np.random.randn(50))
-    ax.plot(x, np.sin(x) + 2 * x + np.random.randn(50))
-    ax.plot(x, np.sin(x) - 0.5 * x + np.random.randn(50))
-    ax.plot(x, np.sin(x) - 2 * x + np.random.randn(50))
-    ax.plot(x, np.sin(x) + np.random.randn(50))
-    ax.set_title("'fivethirtyeight' style sheet")
+ax.plot(x, np.sin(x) + x + np.random.randn(50))
+ax.plot(x, np.sin(x) + 0.5 * x + np.random.randn(50))
+ax.plot(x, np.sin(x) + 2 * x + np.random.randn(50))
+ax.plot(x, np.sin(x) - 0.5 * x + np.random.randn(50))
+ax.plot(x, np.sin(x) - 2 * x + np.random.randn(50))
+ax.plot(x, np.sin(x) + np.random.randn(50))
+ax.set_title("'fivethirtyeight' style sheet")
 
 
 plt.show()
