Commit e0e7de34 authored by Atreya Majumdar's avatar Atreya Majumdar Committed by atreyamaj
Browse files

Removed one hot formal charge

parent 909f6d37
Loading
Loading
Loading
Loading
+2 −12
Original line number Diff line number Diff line
@@ -56,18 +56,8 @@ class MATFeaturizer(MolecularFeaturizer):
  This class requires RDKit to be installed.
  """

  def __init__(
      self,
      one_hot_formal_charge: bool = True,
  ):
    """
    Parameters
    ----------
    one_hot_formal_charge: bool, default True
      If True, formal charges on atoms are one-hot encoded.
    """

    self.one_hot_formal_charge = one_hot_formal_charge
  def __init__(self):
    pass

  def construct_mol(self, mol: RDKitMol) -> RDKitMol:
    """