fix when hit.sum_probs is none (#137)

29f11deb · shenggan · GitHub · a14f6962 · 29f11deb
Unverified Commit 29f11deb authored Jan 15, 2023 by shenggan Committed by GitHub Jan 15, 2023
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

fastfold/data/templates.py fastfold/data/templates.py +4 -1

No files found.
--- a/fastfold/data/templates.py
+++ b/fastfold/data/templates.py
@@ -866,6 +866,9 @@ def _process_single_hit(
            kalign_binary_path=kalign_binary_path,
            _zero_center_positions=_zero_center_positions,
        )
+        if hit.sum_probs is None:
+            features['template_sum_probs'] = [0]
+        else:
            features["template_sum_probs"] = [hit.sum_probs]

        # It is possible there were some errors when parsing the other chains in the