diff --git a/docs/reference/datasets/generation/reaction.md b/docs/reference/datasets/generation/reaction.md index e69de29bb2..41efbecb11 100644 --- a/docs/reference/datasets/generation/reaction.md +++ b/docs/reference/datasets/generation/reaction.md @@ -0,0 +1 @@ +::: beignet.datasets.USPTODataset diff --git a/src/beignet/datasets/_uspto_dataset.py b/src/beignet/datasets/_uspto_dataset.py index 4970ab0cbc..47cefb192b 100644 --- a/src/beignet/datasets/_uspto_dataset.py +++ b/src/beignet/datasets/_uspto_dataset.py @@ -34,11 +34,11 @@ def __init__( super().__init__( root=root, download=download, - identifier=0, - suffix="", - checksum="", - x_keys=[""], - y_keys=[""], + identifier=4171642, + suffix="csv", + checksum="md5:818b06cd7dff707d5ae2c82109ff8668", + x_keys=["reactant"], + y_keys=["product"], transform=transform, target_transform=target_transform, )