Commit 898e6758 authored by Karl Leswing's avatar Karl Leswing
Browse files

Fix SDFLoader constructor args

parent da8419d5
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -240,8 +240,8 @@ class SDFLoader(DataLoader):
  Handles loading of SDF files.
  """

  def __init__(self, clean_mols=False, **kwargs):
    super(SDFLoader, self).__init__(**kwargs)
  def __init__(self, tasks, clean_mols=False, **kwargs):
    super(SDFLoader, self).__init__(tasks, **kwargs)
    self.clean_mols = clean_mols
    self.smiles_field = "smiles"
    self.mol_field = "mol"