Trying to make loss function safe from infinity value - getting weird error: TypeError: unsupported operand type(s) for -: 'int' and 'NoneType'

Question

Original function:

def _build_adversarial(self):
        
        #construct VAE
        # For vae all layers are trainable
        self.set_trainable(self.encode