remove redundant label_number property

- #47 (comment)

Author: aditya0by0

chebai/preprocessing/datasets/base.py

@@ -525,18 +525,6 @@ def raw_file_names_dict(self) -> dict:
         """
         raise NotImplementedError
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the number of labels.
-
-        This property should be implemented by subclasses to provide the number of labels.
-
-        Returns:
-            int: The number of labels. Returns -1 for seq2seq encoding.
-        """
-        raise NotImplementedError
-
 
 class MergedDataset(XYBaseDataModule):
     MERGED = []
@@ -673,13 +661,6 @@ def processed_file_names(self) -> List[str]:
         """
         return ["test.pt", "train.pt", "validation.pt"]
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the number of labels from the first subset.
-        """
-        return self.subsets[0].label_number
-
     @property
     def limits(self):
         """

chebai/preprocessing/datasets/chebi.py

@@ -89,10 +89,6 @@ def setup_processed(self):
                 os.path.join(self.processed_dir, f"{k}.pt"),
             )
 
-    @property
-    def label_number(self):
-        return 500
-
 
 class JCIData(JCIBase):
     READER = dr.OrdReader
@@ -546,10 +542,6 @@ def raw_file_names_dict(self) -> dict:
 
 class JCIExtendedBase(_ChEBIDataExtractor):
 
-    @property
-    def label_number(self):
-        return 500
-
     @property
     def _name(self):
         return "JCI_extended"
@@ -574,16 +566,6 @@ class ChEBIOverX(_ChEBIDataExtractor):
     READER: dr.ChemDataReader = dr.ChemDataReader
     THRESHOLD: int = None
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the number of labels in the dataset.
-
-        Returns:
-            int: The number of labels.
-        """
-        return 854
-
     @property
     def _name(self) -> str:
         """
@@ -676,17 +658,6 @@ class ChEBIOver100(ChEBIOverX):
 
     THRESHOLD: int = 100
 
-    def label_number(self) -> int:
-        """
-        Returns the number of labels in the dataset.
-
-        Overrides the base class method to return the correct number of labels for this threshold.
-
-        Returns:
-            int: The number of labels.
-        """
-        return 854
-
 
 class ChEBIOver50(ChEBIOverX):
     """
@@ -700,17 +671,6 @@ class ChEBIOver50(ChEBIOverX):
 
     THRESHOLD: int = 50
 
-    def label_number(self) -> int:
-        """
-        Returns the number of labels in the dataset.
-
-        Overrides the base class method to return the correct number of labels for this threshold.
-
-        Returns:
-            int: The number of labels.
-        """
-        return 1332
-
 
 class ChEBIOver100DeepSMILES(ChEBIOverXDeepSMILES, ChEBIOver100):
     """

chebai/preprocessing/datasets/pubchem.py

@@ -693,13 +693,6 @@ class PubchemChem(PubChem):
 
     READER: Type[dr.ChemDataReader] = dr.ChemDataReader
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the label number.
-        """
-        return -1
-
 
 class PubchemBPE(PubChem):
     """
@@ -713,13 +706,6 @@ class PubchemBPE(PubChem):
 
     READER: Type[dr.ChemBPEReader] = dr.ChemBPEReader
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the label number.
-        """
-        return -1
-
 
 class SWJChem(SWJPreChem):
     """
@@ -733,13 +719,6 @@ class SWJChem(SWJPreChem):
 
     READER: Type[dr.ChemDataUnlabeledReader] = dr.ChemDataUnlabeledReader
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the label number.
-        """
-        return -1
-
 
 class SWJBPE(SWJPreChem):
     """
@@ -753,13 +732,6 @@ class SWJBPE(SWJPreChem):
 
     READER: Type[dr.ChemBPEReader] = dr.ChemBPEReader
 
-    @property
-    def label_number(self) -> int:
-        """
-        Returns the label number.
-        """
-        return -1
-
 
 class PubChemTokens(PubChem):
     """

chebai/preprocessing/datasets/tox21.py

@@ -39,11 +39,6 @@ def _name(self) -> str:
         """Returns the name of the dataset."""
         return "Tox21MN"
 
-    @property
-    def label_number(self) -> int:
-        """Returns the number of labels."""
-        return 12
-
     @property
     def raw_file_names(self) -> List[str]:
         """Returns a list of raw file names."""
@@ -177,11 +172,6 @@ def _name(self) -> str:
         """Returns the name of the dataset."""
         return "Tox21Chal"
 
-    @property
-    def label_number(self) -> int:
-        """Returns the number of labels."""
-        return 12
-
     @property
     def raw_file_names(self) -> List[str]:
         """Returns a list of raw file names."""