minor changes

Author: aditya0by0

chebai/preprocessing/datasets/base.py

@@ -155,6 +155,8 @@ def _name(self) -> str:
     def _filter_labels(self, row: dict) -> dict:
         """
         Filter labels based on `label_filter`.
+        This method selects specific labels from the `labels` list within the row dictionary
+        according to the index or indices provided by the `label_filter` attribute of the class.
 
         Args:
             row (dict): A dictionary containing the row data.

chebai/preprocessing/datasets/chebi.py

@@ -754,7 +754,7 @@ def processed_dir_main(self) -> str:
             "processed",
         )
 
-    def extract_class_hierarchy(self, chebi_path: str) -> nx.DiGraph:
+    def _extract_class_hierarchy(self, chebi_path: str) -> nx.DiGraph:
         """
         Extracts a subset of ChEBI based on subclasses of the top class ID.