Skip to content

Commit 03b3d3b

Browse files
authored
Merge pull request #9 from ChEB-AI/fix/load_processed_data_from_file
use load_processed_data_from_file
2 parents 3e98f7e + 56ef24b commit 03b3d3b

File tree

1 file changed

+2
-3
lines changed
  • chebai_graph/preprocessing/datasets

1 file changed

+2
-3
lines changed

chebai_graph/preprocessing/datasets/chebi.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -170,9 +170,8 @@ def _merge_props_into_base(self, row):
170170
molecule_attr=molecule_attr,
171171
)
172172

173-
def load_processed_data(self, kind: str = None, filename: str = None):
174-
"""Combine base data set with property values for atoms and bonds."""
175-
base_data = super().load_processed_data(kind, filename)
173+
def load_processed_data_from_file(self, filename):
174+
base_data = super().load_processed_data_from_file(filename)
176175
base_df = pd.DataFrame(base_data)
177176
for property in self.properties:
178177
property_data = torch.load(

0 commit comments

Comments
 (0)