hard code template_featurizer to be hmmsearch for now just to run the unittest

226d6011 · Geoffrey Yu · 31c976b5 · 226d6011
Commit 226d6011 authored Jul 07, 2023 by Geoffrey Yu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

openfold/data/data_modules.py openfold/data/data_modules.py +2 -1

No files found.
--- a/openfold/data/data_modules.py
+++ b/openfold/data/data_modules.py
@@ -151,7 +151,8 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
            chain: i for i, chain in enumerate(self._chain_ids)
        }
-        template_featurizer = templates.TemplateHitFeaturizer(
+        # changed template_featurizer to hmmsearch for now just to run the test
+        template_featurizer = templates.HmmsearchHitFeaturizer(
            mmcif_dir=template_mmcif_dir,
            max_template_date=max_template_date,
            max_hits=max_template_hits,