Check for none and return none
This commit is contained in:
+2
-1
@@ -45,8 +45,9 @@ class PlotMethods(object):
|
|||||||
plot_method = getattr(self, kind)
|
plot_method = getattr(self, kind)
|
||||||
fig = plot_method(table)
|
fig = plot_method(table)
|
||||||
|
|
||||||
if filename:
|
if filename is not None:
|
||||||
fig.savefig(filename)
|
fig.savefig(filename)
|
||||||
|
return None
|
||||||
|
|
||||||
return fig
|
return fig
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user