Update src/preprocess.py

src/preprocess.py

@@ -9,7 +9,7 @@ from torch_geometric.utils import from_rdmol
 from datasets import load_dataset
 
 def get_tox21_split(token, cvfold=None):
-    ds = load_dataset("tschouis/tox21", token=token)
+    ds = load_dataset("ml-jku/tox21", token=token)
     
     train_df = ds["train"].to_pandas()
     val_df = ds["validation"].to_pandas()