Fix syntax error in inline alignment.py template
Browse files- adapter.py +16 -4
adapter.py
CHANGED
|
@@ -101,15 +101,27 @@ class AbAlignment:
|
|
| 101 |
return numbered_seqs, seqs, number_alignment
|
| 102 |
|
| 103 |
def align_encodings(self, encodings, numbered_seqs, seqs, number_alignment):
|
| 104 |
-
aligned_encodings =
|
| 105 |
-
|
|
|
|
|
|
|
| 106 |
|
| 107 |
def reformat_subsets(self, subset_list, mode='seqcoding', align=False, numbered_seqs=None, seqs=None, number_alignment=None):
|
| 108 |
if mode in ['seqcoding', 'restore', 'pseudo_log_likelihood', 'confidence']:
|
| 109 |
return np.concatenate(subset_list)
|
| 110 |
elif align:
|
| 111 |
-
|
| 112 |
-
subset
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 113 |
return aligned_results(
|
| 114 |
aligned_seqs=[''.join(alist) for alist in subset[:,:,-1]],
|
| 115 |
aligned_embeds=subset[:,:,:-1].astype(float),
|
|
|
|
| 101 |
return numbered_seqs, seqs, number_alignment
|
| 102 |
|
| 103 |
def align_encodings(self, encodings, numbered_seqs, seqs, number_alignment):
|
| 104 |
+
aligned_encodings = []
|
| 105 |
+
for res_embed, numbered_seq, seq in zip(encodings, numbered_seqs, seqs):
|
| 106 |
+
aligned_encodings.append(create_alignment(res_embed, numbered_seq, seq, number_alignment))
|
| 107 |
+
return np.concatenate([aligned_encodings], axis=0)
|
| 108 |
|
| 109 |
def reformat_subsets(self, subset_list, mode='seqcoding', align=False, numbered_seqs=None, seqs=None, number_alignment=None):
|
| 110 |
if mode in ['seqcoding', 'restore', 'pseudo_log_likelihood', 'confidence']:
|
| 111 |
return np.concatenate(subset_list)
|
| 112 |
elif align:
|
| 113 |
+
aligned_subsets = []
|
| 114 |
+
for num, subset in enumerate(subset_list):
|
| 115 |
+
start_idx = num * len(subset)
|
| 116 |
+
end_idx = (num + 1) * len(subset)
|
| 117 |
+
aligned_subset = self.align_encodings(
|
| 118 |
+
subset,
|
| 119 |
+
numbered_seqs[start_idx:end_idx],
|
| 120 |
+
seqs[start_idx:end_idx],
|
| 121 |
+
number_alignment
|
| 122 |
+
)
|
| 123 |
+
aligned_subsets.append(aligned_subset)
|
| 124 |
+
subset = np.concatenate(aligned_subsets)
|
| 125 |
return aligned_results(
|
| 126 |
aligned_seqs=[''.join(alist) for alist in subset[:,:,-1]],
|
| 127 |
aligned_embeds=subset[:,:,:-1].astype(float),
|