Commit 8a58f7d4 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Fix type signature

parent 8db9166b
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -83,7 +83,7 @@ def load_data(input_files: List[str],
def load_sdf_files(input_files: List[str],
                   clean_mols: bool = True,
                   tasks: List[str] = [],
                   shard_size: Optional[int] = None) -> List[pd.DataFrame]:
                   shard_size: Optional[int] = None) -> Iterator[pd.DataFrame]:
  """Load SDF file into dataframe.

  Parameters