hemantn commited on
Commit
13d4401
·
1 Parent(s): 2a109cb

Fix syntax error in alignment.py - simplify complex list comprehension

Browse files
Files changed (1) hide show
  1. alignment.py +6 -7
alignment.py CHANGED
@@ -25,13 +25,12 @@ class AbAlignment:
25
 
26
  def align_encodings(self, encodings, numbered_seqs, seqs, number_alignment):
27
 
28
- aligned_encodings = np.concatenate(
29
- [[
30
- create_alignment(
31
- res_embed, numbered_seq, seq, number_alignment
32
- ) for res_embed, numbered_seq, seq in zip(encodings, numbered_seqs, seqs)
33
- ]], axis=0
34
- )
35
  return aligned_encodings
36
 
37
 
 
25
 
26
  def align_encodings(self, encodings, numbered_seqs, seqs, number_alignment):
27
 
28
+ aligned_list = [
29
+ create_alignment(
30
+ res_embed, numbered_seq, seq, number_alignment
31
+ ) for res_embed, numbered_seq, seq in zip(encodings, numbered_seqs, seqs)
32
+ ]
33
+ aligned_encodings = np.concatenate([aligned_list], axis=0)
 
34
  return aligned_encodings
35
 
36