Reverted energy in DFIRE with restraints

Author: brianjimenez

lightdock/scoring/dfire/driver.py

@@ -180,12 +180,9 @@ def __call__(self, receptor, receptor_coordinates, ligand, ligand_coordinates):
                                                                        self.potential.dfire_energy,
                                                                        interface_cutoff=self.cutoff)
 
-        if len(receptor.restraints) and not len(ligand.restraints):
-            return energy
-        else:
-            perc_receptor_restraints = ScoringFunction.restraints_satisfied(receptor.restraints, interface_receptor)
-            perc_ligand_restraints = ScoringFunction.restraints_satisfied(ligand.restraints, interface_ligand)
-            return energy + perc_receptor_restraints * energy + perc_ligand_restraints * energy
+        perc_receptor_restraints = ScoringFunction.restraints_satisfied(receptor.restraints, interface_receptor)
+        perc_ligand_restraints = ScoringFunction.restraints_satisfied(ligand.restraints, interface_ligand)
+        return energy + perc_receptor_restraints * energy + perc_ligand_restraints * energy
 
 
 # Needed to dynamically load the scoring functions from command line