"...src/http/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "113f4d91259ecc01933f6732d8915a7ae86cacb4"
Commit a6a467e0 authored by Sachin Kadyan's avatar Sachin Kadyan
Browse files

Added passing of sequence embedding mode flag from `data_modules` to...

Added passing of sequence embedding mode flag from `data_modules` to `data_pipeline` for training and inference pipelines.
- Passing the config.data.seqemb_mode.enabled flag to the FASTA, PDB, and MMCIF data pipelines.
parent 2ba07feb
...@@ -186,7 +186,8 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset): ...@@ -186,7 +186,8 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
mmcif=mmcif_object, mmcif=mmcif_object,
alignment_dir=alignment_dir, alignment_dir=alignment_dir,
chain_id=chain_id, chain_id=chain_id,
alignment_index=alignment_index alignment_index=alignment_index,
seqemb_mode=self.config.seqemb_mode.enabled
) )
return data return data
...@@ -251,6 +252,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset): ...@@ -251,6 +252,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
chain_id=chain_id, chain_id=chain_id,
alignment_index=alignment_index, alignment_index=alignment_index,
_structure_index=structure_index, _structure_index=structure_index,
seqemb_mode=self.config.seqemb_mode.enabled,
) )
else: else:
raise ValueError("Extension branch missing") raise ValueError("Extension branch missing")
...@@ -260,6 +262,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset): ...@@ -260,6 +262,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
fasta_path=path, fasta_path=path,
alignment_dir=alignment_dir, alignment_dir=alignment_dir,
alignment_index=alignment_index, alignment_index=alignment_index,
seqemb_mode=self.config.seqemb_mode.enabled,
) )
if(self._output_raw): if(self._output_raw):
......
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