Skip to content

Commit 31b12db

Browse files
committed
Revert "fix reader ident error"
This reverts commit 517a5a2.
1 parent 6a38317 commit 31b12db

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chebai/preprocessing/reader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def _get_raw_label(self, row: Dict[str, Any]) -> Any:
5151

5252
def _get_raw_id(self, row: Dict[str, Any]) -> Any:
5353
"""Get raw ID from the row."""
54-
return row.get("ident", row["id"])
54+
return row.get("ident", row["features"])
5555

5656
def _get_raw_group(self, row: Dict[str, Any]) -> Any:
5757
"""Get raw group from the row."""

0 commit comments

Comments
 (0)