datasets

Genentech · Jul 17, 2024 · 069d2e4 · 069d2e4
1 parent dfc6fca
commit 069d2e4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -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,
         )