Fix issue with stat files

Fluorescence-Tools · Apr 18, 2024 · 98ff783 · 98ff783
1 parent d31045d
commit 98ff783
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pyext/src/restraints/AVNetworkRestraint.py b/pyext/src/restraints/AVNetworkRestraint.py
@@ -178,6 +178,15 @@ def evaluate(self):
         """Evaluate the score of the restraint."""
         return self.rs.unprotected_evaluate(None) * self.weight
 
+    def get_output(self):
+        self.mdl.update()
+        output = {}
+        score = self.weight * self.rs.unprotected_evaluate(None)
+        output["_TotalScore"] = str(score)
+        output["AVNetworkRestraint_" +
+               self.label] = str(score)
+        return output
+
     def add_to_model(self, add_to_rmf=True):
         IMP.pmi.tools.add_restraint_to_model(self.mdl, self.rs,
                                              add_to_rmf=add_to_rmf)