Unverified Commit 46f62b57 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #2319 from HFooladi/patch1 

changing the expected output of get_task_names for DiskDataset to List[str]
parents 355954b3 e685470c

deepchem/data/datasets.py

+1 −1
Original line number Diff line number Diff line
@@ -1410,7 +1410,7 @@ class DiskDataset(Dataset): 
    shutil.copytree(self.data_dir, new_data_dir)

    return DiskDataset(new_data_dir)



  def get_task_names(self) -> np.ndarray:

  def get_task_names(self) -> List[str]:

    """Gets learning tasks associated with this dataset."""

    return self.tasks

CRA Git | Maintained and supported by SUSTech CRA and CCSE