hemantn commited on
Commit
f07c732
·
1 Parent(s): 960915a

Fix syntax error in inline alignment.py template

Browse files
Files changed (1) hide show
  1. 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 = np.concatenate([[[create_alignment(res_embed, numbered_seq, seq, number_alignment) for res_embed, numbered_seq, seq in zip(encodings, numbered_seqs, seqs)]], axis=0)
105
- return aligned_encodings
 
 
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
- subset_list = [self.align_encodings(subset, numbered_seqs[num*len(subset):(num+1)*len(subset)], seqs[num*len(subset):(num+1)*len(subset)], number_alignment) for num, subset in enumerate(subset_list)]
112
- subset = np.concatenate(subset_list)
 
 
 
 
 
 
 
 
 
 
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),