From 85b67e509ecb78955d2566467b74b379eb88c0ad Mon Sep 17 00:00:00 2001
From: Allen Goodman <allen.goodman@icloud.com>
Date: Wed, 17 Jul 2024 15:42:56 -0400
Subject: [PATCH] datasets

---
 src/beignet/datasets/_chen_sabdab_dataset.py | 44 ++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 src/beignet/datasets/_chen_sabdab_dataset.py

diff --git a/src/beignet/datasets/_chen_sabdab_dataset.py b/src/beignet/datasets/_chen_sabdab_dataset.py
new file mode 100644
index 0000000000..7d438e75b3
--- /dev/null
+++ b/src/beignet/datasets/_chen_sabdab_dataset.py
@@ -0,0 +1,44 @@
+from pathlib import Path
+from typing import Callable
+
+from beignet.transforms import Transform
+
+from ._tdc_dataset import TDCDataset
+
+
+class ChenSAbDabDataset(TDCDataset):
+    def __init__(
+        self,
+        root: str | Path,
+        *,
+        download: bool = False,
+        transform: Callable | Transform | None = None,
+        target_transform: Callable | Transform | None = None,
+    ):
+        r"""
+        Parameters
+        ----------
+        root : str | Path
+            Root directory of dataset.
+
+        download: bool
+            If `True`, downloads the dataset to the root directory. If dataset
+            already exists, it is not redownloaded. Default, `False`.
+
+        transform : Callable | Transform | None
+            Transforms the input.
+
+        target_transform : Callable | Transform | None
+            Transforms the target.
+        """
+        super().__init__(
+            root=root,
+            download=download,
+            identifier=4167164,
+            suffix="tsv",
+            checksum="md5:3b58b73e0d5371a2ccc865ca045a2061",
+            x_keys=["X"],
+            y_keys=["Y"],
+            transform=transform,
+            target_transform=target_transform,
+        )