Commit e685470c authored by Hosein Fooladi's avatar Hosein Fooladi
Browse files

changing output of get_task_names for DistDataset to list

parent 355954b3

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