Skip to content

Commit 0c2fca1

Browse files
committed
pd.to_pickle instead of pickle.dump for code consistency
1 parent 9c25543 commit 0c2fca1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chebai/preprocessing/datasets/chebi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def save_raw(self, data: pd.DataFrame, filename: str):
216216
pd.to_pickle(data, open(os.path.join(self.raw_dir, filename), "wb"))
217217

218218
def save_processed(self, data: pd.DataFrame, filename: str):
219-
pickle.dump(data, open(os.path.join(self.processed_dir_main, filename), "wb"))
219+
pd.to_pickle(data, open(os.path.join(self.processed_dir_main, filename), "wb"))
220220

221221
def _load_dict(self, input_file_path):
222222
"""

0 commit comments

Comments
 (0)