def state_dict()

in models/neurvol1.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

    def state_dict(self, destination=None, prefix='', keep_vars=False):
        ret = super(Autoencoder, self).state_dict(destination, prefix, keep_vars)
        if not self.estimatebg:
            for k in self.bg.keys():
                del ret[prefix+"bg."+k]
        return ret