Skip to content

Commit

Permalink
mend
Browse files Browse the repository at this point in the history
  • Loading branch information
amva13 committed Aug 19, 2024
1 parent f076ab9 commit b555583
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tdc/single_pred/mpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
from ..utils import print_sys, fuzzy_search, property_dataset_load
from ..metadata import dataset_names


class MPC(single_pred_dataset.DataLoader):

def __init__(self, name, path="./data"):
Expand All @@ -27,7 +28,9 @@ def get_data(self):
self.data = Data(self.name)
self.data(Descriptors.SMILES)
except:
raise Exception("could not find dataset {}. For list of MoleculeAce datasets see https://github.com/bidd-group/MPCD/tree/main?tab=readme-ov-file#overview-of-the-mpc-benchmark-datasets".format(self.name))
raise Exception(
"could not find dataset {}. For list of MoleculeAce datasets see https://github.com/bidd-group/MPCD/tree/main?tab=readme-ov-file#overview-of-the-mpc-benchmark-datasets"
.format(self.name))
return self.data

def get_split(self):
Expand Down

0 comments on commit b555583

Please sign in to comment.