Add chain_iptm as output

Author: samuelmurail

colabfold/batch.py

@@ -490,6 +490,10 @@ def callback(result, recycles):
                   del pae
                 del plddt
                 json.dump(scores, handle)
+                
+                if "chain_iptm" in result:
+                    scores.update({"chain_iptm": np.around(result["chain_iptm"].astype(float), 4).tolist()})
+                    scores.update({"ptm_matrix": np.around(result["ptm_matrix"].astype(float), 4).tolist()})
 
             del result, unrelaxed_protein