revert _skip_data_methods_on_init

Author: aditya0by0

chebai/preprocessing/datasets/deepGO/protein_pretraining.py

@@ -38,7 +38,7 @@ def __init__(self, **kwargs):
         Args:
             **kwargs: Additional arguments for the superclass initialization.
         """
-        self._go_uniprot_extractor = GOUniProtOver250(_skip_data_methods_on_init=True)
+        self._go_uniprot_extractor = GOUniProtOver250()
         assert self._go_uniprot_extractor.go_branch == GOUniProtOver250._ALL_GO_BRANCHES
 
         self.max_sequence_length: int = int(kwargs.get("max_sequence_length", 1002))

tests/unit/dataset_classes/testChEBIOverX.py

@@ -19,9 +19,7 @@ def setUpClass(cls, mock_makedirs, mock_processed_dir_main: PropertyMock) -> Non
             mock_processed_dir_main (PropertyMock): Mocked property for the processed directory path.
         """
         mock_processed_dir_main.return_value = "/mock/processed_dir"
-        cls.chebi_extractor = ChEBIOverX(
-            chebi_version=231, _skip_data_methods_on_init=True
-        )
+        cls.chebi_extractor = ChEBIOverX(chebi_version=231)
         cls.test_graph = ChebiMockOntology.get_transitively_closed_graph()
 
     @patch("builtins.open", new_callable=mock_open)

tests/unit/dataset_classes/testChebiDataExtractor.py

@@ -35,7 +35,7 @@ def setUpClass(
 
         # Create an instance of the dataset
         cls.extractor: _ChEBIDataExtractor = _ChEBIDataExtractor(
-            chebi_version=231, chebi_version_train=200, _skip_data_methods_on_init=True
+            chebi_version=231, chebi_version_train=200
         )
 
         # Mock instance for _chebi_version_train_obj

tests/unit/dataset_classes/testChebiOverXPartial.py

@@ -16,9 +16,7 @@ def setUpClass(cls, mock_makedirs) -> None:
         """
         Set up the ChEBIOverXPartial instance with a mock processed directory path and a test graph.
         """
-        cls.chebi_extractor = ChEBIOverXPartial(
-            top_class_id=11111, chebi_version=231, _skip_data_methods_on_init=True
-        )
+        cls.chebi_extractor = ChEBIOverXPartial(top_class_id=11111, chebi_version=231)
         cls.test_graph = ChebiMockOntology.get_transitively_closed_graph()
 
     @patch(

tests/unit/dataset_classes/testDynamicDataset.py

@@ -38,7 +38,7 @@ def setUpClass(
         _DynamicDataset.READER = ReaderMock
 
         # Creating an instance of the dataset
-        cls.dataset: _DynamicDataset = _DynamicDataset(_skip_data_methods_on_init=True)
+        cls.dataset: _DynamicDataset = _DynamicDataset()
 
         # Dataset with a balanced distribution of labels
         X = [

tests/unit/dataset_classes/testGOUniProDataExtractor.py

@@ -35,7 +35,7 @@ def setUpClass(
 
         _GOUniProtDataExtractor.READER = ProteinDataReader
 
-        cls.extractor = _GOUniProtDataExtractor(_skip_data_methods_on_init=True)
+        cls.extractor = _GOUniProtDataExtractor()
 
     def test_term_callback(self) -> None:
         """

tests/unit/dataset_classes/testGoUniProtOverX.py

@@ -17,7 +17,7 @@ def setUpClass(cls, mock_makedirs) -> None:
         """
         Set up the class for tests by initializing the extractor, graph, and input DataFrame.
         """
-        cls.extractor = _GOUniProtOverX(_skip_data_methods_on_init=True)
+        cls.extractor = _GOUniProtOverX()
         cls.test_graph: nx.DiGraph = GOUniProtMockData.get_transitively_closed_graph()
         cls.input_df: pd.DataFrame = GOUniProtMockData.get_data_in_dataframe().iloc[
             :, :4

tests/unit/dataset_classes/testProteinPretrainingData.py

@@ -37,7 +37,7 @@ def setUpClass(
         _ProteinPretrainingData.READER = ProteinDataReader
 
         # Initialize the extractor instance
-        cls.extractor = _ProteinPretrainingData(_skip_data_methods_on_init=True)
+        cls.extractor = _ProteinPretrainingData()
 
     @patch(
         "builtins.open",

tests/unit/dataset_classes/testTox21Challenge.py

@@ -24,7 +24,7 @@ def setUpClass(cls, mock_makedirs) -> None:
         This is run once for the test class.
         """
         Tox21Challenge.READER = ChemDataReader
-        cls.tox21 = Tox21Challenge(_skip_data_methods_on_init=True)
+        cls.tox21 = Tox21Challenge()
 
     @patch("rdkit.Chem.SDMolSupplier")
     def test_load_data_from_file(self, mock_sdmol_supplier: patch) -> None:

tests/unit/dataset_classes/testXYBaseDataModule.py

@@ -30,7 +30,6 @@ def setUpClass(cls, mock_makedirs, mock_name_property: PropertyMock) -> None:
         cls.module = XYBaseDataModule(
             label_filter=1,  # Provide a label_filter
             balance_after_filter=1.0,  # Balance ratio
-            _skip_data_methods_on_init=True,
         )
 
     def test_filter_labels_valid_index(self) -> None: