Skip to content

Commit 517a5a2

Browse files
committed
fix reader ident error
1 parent ae47608 commit 517a5a2

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["features"])
54+
return row.get("ident", row["id"])
5555

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

0 commit comments

Comments
 (0)