[plots] return CS in contour (#81)

flav-io · Mar 15, 2019 · 47c3581 · 47c3581
1 parent 48347a1
commit 47c3581
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flavio/plots/plotfunctions.py b/flavio/plots/plotfunctions.py
@@ -579,6 +579,7 @@ def contour(x, y, z, levels,
     CS = ax.contour(x, y, z, levels=levels, **_contour_args)
     if label is not None:
         CS.collections[0].set_label(label)
+    return CS
 
 
 def flavio_branding(x=0.8, y=0.94, version=True):