Merge dd6ace0 into 284cead

zblz · Nov 27, 2018 · 2affe8b · 2affe8b
2 parents 284cead + dd6ace0
commit 2affe8b
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/naima/plot.py b/naima/plot.py
@@ -1417,18 +1417,20 @@ def plot_distribution(samples, label, figure=None):
 
     histnbins = min(max(25, int(len(samples) / 100.0)), 100)
     xlabel = "" if label is None else label
+
+    if isinstance(samples, u.Quantity):
+        samples_nounit = samples.value
+    else:
+        samples_nounit = samples
+
     n, x, _ = ax.hist(
-        samples,
+        samples_nounit,
         histnbins,
         histtype="stepfilled",
         color=color_cycle[0],
         lw=0,
         density=True,
     )
-    if isinstance(samples, u.Quantity):
-        samples_nounit = samples.value
-    else:
-        samples_nounit = samples
 
     kde = stats.kde.gaussian_kde(samples_nounit)
     ax.plot(x, kde(x), color="k", label="KDE")