Commit 0d348792 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

fixed the error when loading a monomeric training object

parent caac73c0
...@@ -434,7 +434,6 @@ class OpenFoldSingleMultimerDataset(torch.utils.data.Dataset): ...@@ -434,7 +434,6 @@ class OpenFoldSingleMultimerDataset(torch.utils.data.Dataset):
self.multimer_data_pipeline = data_pipeline.DataPipelineMultimer( self.multimer_data_pipeline = data_pipeline.DataPipelineMultimer(
monomer_data_pipeline=self.data_pipeline monomer_data_pipeline=self.data_pipeline
) )
self.feature_pipeline = feature_pipeline.FeaturePipeline(config) self.feature_pipeline = feature_pipeline.FeaturePipeline(config)
def _parse_mmcif(self, path, file_id, chain_id, alignment_dir, alignment_index): def _parse_mmcif(self, path, file_id, chain_id, alignment_dir, alignment_index):
...@@ -470,7 +469,6 @@ class OpenFoldSingleMultimerDataset(torch.utils.data.Dataset): ...@@ -470,7 +469,6 @@ class OpenFoldSingleMultimerDataset(torch.utils.data.Dataset):
def __getitem__(self, idx): def __getitem__(self, idx):
mmcif_id = self.idx_to_mmcif_id(idx) mmcif_id = self.idx_to_mmcif_id(idx)
chains = self.mmcif_data_cache[mmcif_id]['chain_ids'] chains = self.mmcif_data_cache[mmcif_id]['chain_ids']
is_multimer = (len(chains)>1)
seqs = self.mmcif_data_cache[mmcif_id]['seqs'] seqs = self.mmcif_data_cache[mmcif_id]['seqs']
fasta_str = "" fasta_str = ""
for c,s in zip(chains,seqs): for c,s in zip(chains,seqs):
...@@ -481,7 +479,7 @@ class OpenFoldSingleMultimerDataset(torch.utils.data.Dataset): ...@@ -481,7 +479,7 @@ class OpenFoldSingleMultimerDataset(torch.utils.data.Dataset):
# process all_chain_features # process all_chain_features
all_chain_features = self.feature_pipeline.process_features(all_chain_features, all_chain_features = self.feature_pipeline.process_features(all_chain_features,
mode=self.mode, mode=self.mode,
is_multimer=is_multimer) is_multimer=True)
alignment_index = None alignment_index = None
ground_truth=[] ground_truth=[]
......
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