kelu01 commited on
Commit
e27e7eb
·
verified ·
1 Parent(s): 7c72a8a

Update pipe.py

Browse files
Files changed (1) hide show
  1. pipe.py +0 -2
pipe.py CHANGED
@@ -13,9 +13,7 @@ class SmilesDiffusionPipe(Pipeline):
13
  self.voc = voc
14
 
15
  def restore_halogen(self, smiles):
16
- # Replace L → Cl when NOT inside brackets
17
  smiles = re.sub(r'(?<!\[)L(?!\])', 'Cl', smiles)
18
- # Replace R → Br when NOT inside brackets
19
  smiles = re.sub(r'(?<!\[)R(?!\])', 'Br', smiles)
20
  return smiles
21
 
 
13
  self.voc = voc
14
 
15
  def restore_halogen(self, smiles):
 
16
  smiles = re.sub(r'(?<!\[)L(?!\])', 'Cl', smiles)
 
17
  smiles = re.sub(r'(?<!\[)R(?!\])', 'Br', smiles)
18
  return smiles
19