Check for none and return none

pull/121/head
Vinayak Mehta 2020-08-14 02:35:50 +05:30
parent f8b6181988
commit b741c0a9e9
No known key found for this signature in database
GPG Key ID: 2170CDB940114C1D
1 changed files with 2 additions and 1 deletions

View File

@ -45,8 +45,9 @@ class PlotMethods(object):
plot_method = getattr(self, kind)
fig = plot_method(table)
if filename:
if filename is not None:
fig.savefig(filename)
return None
return fig