Update esm_scripts/extract.py
Browse files- esm_scripts/extract.py +1 -1
esm_scripts/extract.py
CHANGED
@@ -139,7 +139,7 @@ def run_demo(protein_name, protein_seq, model, alphabet, include,
|
|
139 |
data_loader = torch.utils.data.DataLoader(
|
140 |
dataset, collate_fn=alphabet.get_batch_converter(truncation_seq_length), batch_sampler=batches
|
141 |
)
|
142 |
-
print(f"Read
|
143 |
|
144 |
# output_dir.mkdir(parents=True, exist_ok=True)
|
145 |
return_contacts = "contacts" in include
|
|
|
139 |
data_loader = torch.utils.data.DataLoader(
|
140 |
dataset, collate_fn=alphabet.get_batch_converter(truncation_seq_length), batch_sampler=batches
|
141 |
)
|
142 |
+
print(f"Read sequences")
|
143 |
|
144 |
# output_dir.mkdir(parents=True, exist_ok=True)
|
145 |
return_contacts = "contacts" in include
|