Fix default value of filename in calc_density_correction

This commit is contained in:
Gernot J. Kraberger 2017-10-24 09:49:54 +02:00
parent 4649b2142c
commit d00575632c
1 changed files with 4 additions and 4 deletions

View File

@ -1362,15 +1362,15 @@ class SumkDFT(object):
"""
assert dm_type in ('vasp', 'wien2k'), "'dm_type' must be either 'vasp' or 'wienk'"
assert type(filename) == StringType, ("calc_density_correction: "
"filename has to be a string!")
if filename is None:
if dm_type == 'wien2k':
filename = 'dens_mat.dat'
elif dm_type == 'vasp':
filename = 'GAMMA'
assert type(filename) == StringType, ("calc_density_correction: "
"filename has to be a string!")
ntoi = self.spin_names_to_ind[self.SO]
spn = self.spin_block_names[self.SO]
dens = {sp: 0.0 for sp in spn}