Unverified Commit 989f2d93 authored by Dingquan Yu's avatar Dingquan Yu Committed by GitHub
Browse files

Update data_pipeline.py

parent 8d8584ac
...@@ -1163,7 +1163,7 @@ class DataPipelineMultimer: ...@@ -1163,7 +1163,7 @@ class DataPipelineMultimer:
chain_fasta_str = f'>{chain_id}\n{sequence}\n' chain_fasta_str = f'>{chain_id}\n{sequence}\n'
if chain_alignment_index is None and not os.path.exists(chain_alignment_dir): if chain_alignment_index is None and not os.path.exists(chain_alignment_dir):
raise ValueError(f"Alignments for {chain_id} not found...") raise ValueError(f"Alignments for {chain_id} not found...")
with temp_fasta_file(chain_fasta_str) as chain_fasta_path: with temp_fasta_file(chain_fasta_str) as chain_fasta_path:
chain_features = self._monomer_data_pipeline.process_fasta( chain_features = self._monomer_data_pipeline.process_fasta(
...@@ -1351,4 +1351,4 @@ class DataPipelineMultimer: ...@@ -1351,4 +1351,4 @@ class DataPipelineMultimer:
# Pad MSA to avoid zero-sized extra_msa. # Pad MSA to avoid zero-sized extra_msa.
np_example = pad_msa(np_example, 512) np_example = pad_msa(np_example, 512)
return np_example return np_example
\ No newline at end of file
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