Skip to content

Commit fc50c31

Browse files
committed
pretrain: set labels to None instead of using new reader
1 parent d8e2efb commit fc50c31

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

chebai/preprocessing/datasets/protein_pretraining.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,13 +171,15 @@ def _load_dict(self, input_file_path: str) -> Generator[Dict[str, Any], None, No
171171
Dict[str, Any]: A dictionary containing:
172172
- `features` (str): The sequence data from the file.
173173
- `ident` (Any): The identifier from row index 0.
174+
- `labels`: Set to None
174175
"""
175176
with open(input_file_path, "rb") as input_file:
176177
df = pd.read_pickle(input_file)
177178
for row in df.values:
178179
yield dict(
179180
features=row[self._DATA_REPRESENTATION_IDX],
180181
ident=row[self._ID_IDX],
182+
labels=None,
181183
)
182184

183185
# ------------------------------ Phase: Dynamic Splits -----------------------------------

0 commit comments

Comments
 (0)