Remove problematic alignment from restore method - return sequences as-is
Browse files- adapter.py +3 -20
adapter.py
CHANGED
|
@@ -600,26 +600,9 @@ class HFAbRestore(AbRestore):
|
|
| 600 |
restored_seq = restored_seq.replace(' ', '')
|
| 601 |
restored_seqs.append(restored_seq)
|
| 602 |
|
| 603 |
-
#
|
| 604 |
-
|
| 605 |
-
|
| 606 |
-
if len(restored_seqs) > 1:
|
| 607 |
-
# Find the maximum length
|
| 608 |
-
max_len = max(len(seq) for seq in restored_seqs)
|
| 609 |
-
# Pad shorter sequences with the last character
|
| 610 |
-
aligned_seqs = []
|
| 611 |
-
for seq in restored_seqs:
|
| 612 |
-
if len(seq) < max_len:
|
| 613 |
-
# Pad with the last character of the sequence
|
| 614 |
-
padded_seq = seq + seq[-1] * (max_len - len(seq))
|
| 615 |
-
aligned_seqs.append(padded_seq)
|
| 616 |
-
else:
|
| 617 |
-
aligned_seqs.append(seq)
|
| 618 |
-
return np.array(aligned_seqs)
|
| 619 |
-
else:
|
| 620 |
-
return restored_seqs[0]
|
| 621 |
-
else:
|
| 622 |
-
return np.array(restored_seqs) if len(restored_seqs) > 1 else restored_seqs[0]
|
| 623 |
|
| 624 |
def add_angle_brackets(seq):
|
| 625 |
# Assumes input is 'VH|VL' or 'VH|' or '|VL'
|
|
|
|
| 600 |
restored_seq = restored_seq.replace(' ', '')
|
| 601 |
restored_seqs.append(restored_seq)
|
| 602 |
|
| 603 |
+
# For now, return restored sequences without alignment
|
| 604 |
+
# Proper ANARCI-based alignment requires full implementation of alignment functions
|
| 605 |
+
return np.array(restored_seqs) if len(restored_seqs) > 1 else restored_seqs[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 606 |
|
| 607 |
def add_angle_brackets(seq):
|
| 608 |
# Assumes input is 'VH|VL' or 'VH|' or '|VL'
|