From 4d449674c308e0fac9697ed4ca83ef1fd0932414 Mon Sep 17 00:00:00 2001
From: karinazad <karina.zadorozhny@gmail.com>
Date: Wed, 22 Jan 2025 12:54:36 -0500
Subject: [PATCH 1/2] fix y keys

---
 src/beignet/datasets/_chembl_dataset.py | 2 +-
 src/beignet/datasets/_tdc_dataset.py    | 3 ++-
 src/beignet/datasets/_zinc_dataset.py   | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/beignet/datasets/_chembl_dataset.py b/src/beignet/datasets/_chembl_dataset.py
index 1e9b10ab71..fd11887b50 100644
--- a/src/beignet/datasets/_chembl_dataset.py
+++ b/src/beignet/datasets/_chembl_dataset.py
@@ -34,6 +34,6 @@ def __init__(
             suffix="csv",
             checksum="md5:b0cedfe468c6331487f7084236944bac",
             x_keys=["smiles"],
-            y_keys=[],
+            y_keys=None,
             transform=transform,
         )
diff --git a/src/beignet/datasets/_tdc_dataset.py b/src/beignet/datasets/_tdc_dataset.py
index 28cfca1cf6..cd551b5155 100644
--- a/src/beignet/datasets/_tdc_dataset.py
+++ b/src/beignet/datasets/_tdc_dataset.py
@@ -26,6 +26,7 @@ def __init__(
         y_keys: List[str] | None = None,
         transform: Callable | Transform | None = None,
         target_transform: Callable | Transform | None = None,
+        sep="\t",
     ):
         super().__init__()
 
@@ -45,7 +46,7 @@ def __init__(
 
         match path.suffix:
             case ".csv":
-                self._data = pandas.read_csv(path)
+                self._data = pandas.read_csv(path, sep=None)
             case ".pkl":
                 self._data = pandas.read_pickle(path)
             case ".tab" | ".tsv":
diff --git a/src/beignet/datasets/_zinc_dataset.py b/src/beignet/datasets/_zinc_dataset.py
index 174687c81b..d50969a33c 100644
--- a/src/beignet/datasets/_zinc_dataset.py
+++ b/src/beignet/datasets/_zinc_dataset.py
@@ -34,6 +34,6 @@ def __init__(
             suffix="tsv",
             checksum="md5:9e4754d72db297d496def3498a926979",
             x_keys=["smiles"],
-            y_keys=[],
+            y_keys=None,
             transform=transform,
         )

From b6f10a6a40b212d8af4e37e73cea227422f547a2 Mon Sep 17 00:00:00 2001
From: karinazad <karina.zadorozhny@gmail.com>
Date: Wed, 22 Jan 2025 13:15:17 -0500
Subject: [PATCH 2/2] sep

---
 src/beignet/datasets/_tdc_dataset.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/beignet/datasets/_tdc_dataset.py b/src/beignet/datasets/_tdc_dataset.py
index cd551b5155..82a8d47536 100644
--- a/src/beignet/datasets/_tdc_dataset.py
+++ b/src/beignet/datasets/_tdc_dataset.py
@@ -26,7 +26,6 @@ def __init__(
         y_keys: List[str] | None = None,
         transform: Callable | Transform | None = None,
         target_transform: Callable | Transform | None = None,
-        sep="\t",
     ):
         super().__init__()