Skip to content

GitLab

Commit 226d6011 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

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

parent 31c976b5
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
openfold/data/data_modules.py
View file @ 226d6011
...@@ -151,7 +151,8 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset): ...@@ -151,7 +151,8 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
chain: i for i, chain in enumerate(self._chain_ids) 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, mmcif_dir=template_mmcif_dir,
max_template_date=max_template_date, max_template_date=max_template_date,
max_hits=max_template_hits, max_hits=max_template_hits,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment