hemantn commited on
Commit
55a4139
·
1 Parent(s): de33042

Remove problematic alignment from restore method - return sequences as-is

Browse files
Files changed (1) hide show
  1. 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
- # Handle alignment if requested
604
- if align:
605
- # Simple alignment: ensure all sequences have the same length by padding
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'