Skip to content

Commit 74ba35c

Browse files
committed
update dataset load output and add zinc250k
1 parent 3984b9a commit 74ba35c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ smiles_list, property_np_array = molecular_data.data, molecular_data.target
9797

9898
# the target could be None if loading an unlabeled dataset
9999
from torch_molecule.datasets import load_zinc250k
100-
molecular_data = load_zinc250k(local_dir='torchmol_data', target_cols=target_cols)
100+
molecular_data = load_zinc250k(local_dir='torchmol_data')
101101
smiles_list = molecular_data.data
102102
assert molecular_data.target is None
103103
```

0 commit comments

Comments
 (0)