Remove debug statements and angle brackets formatting - restore functionality complete

adapter.py

@@ -571,7 +571,6 @@ class HFAbRestore(AbRestore):
         
         if align:
             # Implement alignment using ANARCI to create spread sequences
-            print("DEBUG: Using alignment to create spread sequences...")
             seqs = self._sequence_aligning(seqs)
             nr_seqs = len(seqs)//self.spread
             
@@ -591,7 +590,6 @@ class HFAbRestore(AbRestore):
             predictions = predictions[range(predictions.shape[0]), best_seq_idx]
             seqs = np.take_along_axis(seqs, best_seq_idx.view(-1, 1).cpu().numpy(), axis=1)
         else:
-            print(f"DEBUG: Processing sequences without alignment: {seqs}")
             tokens = self.tokenizer(seqs, pad=True, w_extra_tkns=False, device=self.used_device)
             predictions = self.AbLang(tokens)[:,:,1:21]
 
@@ -605,9 +603,7 @@ class HFAbRestore(AbRestore):
             seqs = [f"{h}|{l}" for h,l in zip(seqs[:n_seqs], seqs[n_seqs:])]
         
         from extra_utils import res_to_seq
-        result = np.array([res_to_seq(seq, 'restore') for seq in np.c_[restored_seqs, np.vectorize(len)(seqs)]])
-        print(f"DEBUG: Final result: {result}")
-        return result
+        return np.array([res_to_seq(seq, 'restore') for seq in np.c_[restored_seqs, np.vectorize(len)(seqs)]])
     
     def _sequence_aligning(self, seqs):
         """Create spread sequences using ANARCI alignment."""
@@ -1027,11 +1023,6 @@ class AbLang2PairedHuggingFaceAdapter(AbEncoding, AbRestore, AbAlignment, AbScor
     def restore(self, seqs, align=False, **kwargs):
         hf_abrestore = HFAbRestore(self.AbLang, self.tokenizer, spread=self.spread, device=self.used_device, ncpu=self.ncpu)
         restored = hf_abrestore.restore(seqs, align=align)
-        # Apply angle brackets formatting
-        if isinstance(restored, np.ndarray):
-            restored = np.array([add_angle_brackets(seq) for seq in restored])
-        else:
-            restored = [add_angle_brackets(seq) for seq in restored]
         return restored
 
 def extract_input_ids(tokens, device):