Address template edge case

236c6865 · Gustaf Ahdritz · b9eede45 · 236c6865
Commit 236c6865 authored Jul 19, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

openfold/data/data_pipeline.py openfold/data/data_pipeline.py +6 -3

No files found.
--- a/openfold/data/data_pipeline.py
+++ b/openfold/data/data_pipeline.py
@@ -100,9 +100,12 @@ def unify_template_features(
        if(n_templates != 0):
            out_dicts.append(out_dict)

-    out_dict = {
-        k: np.concatenate([od[k] for od in out_dicts]) for k in out_dicts[0]
-    }
+    if(len(out_dicts) > 0):
+        out_dict = {
+            k: np.concatenate([od[k] for od in out_dicts]) for k in out_dicts[0]
+        }
+    else:
+        out_dict = empty_template_feats(sum(seq_lens))

    return out_dict