Commit bebb052a authored by seyonechithrananda's avatar seyonechithrananda
Browse files

change init

parent 80f84db6
Loading
Loading
Loading
Loading
+4 −1
Original line number Diff line number Diff line
@@ -42,7 +42,10 @@ class RobertaFeaturizer(RobertaTokenizerFast, MolecularFeaturizer):
  #def __init__(self, **kwargs):
  def __init__(self, input_ids, attention_mask):
    # super().__init__(**kwargs)
    super().__init__(input_ids, attention_mask)
    #super().__init__(input_ids, attention_mask)
    self.input_ids = input_ids
    self.attention_mask = attention_mask
    
    return