Skip to content

Commit 3615fb1

Browse files
committed
Merge branch 'dev' into fix/directed-to-undirected-graph
2 parents b1f2da3 + 03b3d3b commit 3615fb1

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
@@ -176,9 +176,8 @@ def _merge_props_into_base(self, row):
176176
molecule_attr=molecule_attr,
177177
)
178178

179-
def load_processed_data(self, kind: str = None, filename: str = None):
180-
"""Combine base data set with property values for atoms and bonds."""
181-
base_data = super().load_processed_data(kind, filename)
179+
def load_processed_data_from_file(self, filename):
180+
base_data = super().load_processed_data_from_file(filename)
182181
base_df = pd.DataFrame(base_data)
183182
for property in self.properties:
184183
property_data = torch.load(

0 commit comments

Comments
 (0)