diff --git a/colabfold/batch.py b/colabfold/batch.py index 49f925e8..37d55906 100644 --- a/colabfold/batch.py +++ b/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