Skip to content

Commit 97e99fc

Browse files
committed
Update chebi.py
1 parent 39f3e9f commit 97e99fc

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
@@ -897,7 +897,7 @@ def _extract_class_hierarchy(self, chebi_path: str) -> nx.DiGraph:
897897
)
898898
return g
899899

900-
def select_classes(self, g: nx.DiGraph, *args, **kwargs) -> List:
900+
def select_classes(self, g: nx.DiGraph, *args, **kwargs) -> list:
901901
"""Only selects classes that meet the threshold AND are subclasses of the top class ID (including itself)."""
902902
smiles = nx.get_node_attributes(g, "smiles")
903903
nodes = list(

0 commit comments

Comments
 (0)