no splits file for aug data - it must use org splits

Author: aditya0by0

chebai/preprocessing/datasets/base.py

@@ -1010,8 +1010,7 @@ def _generate_dynamic_splits(self) -> None:
 
         combined_split_assignment = pd.concat(split_assignment_list, ignore_index=True)
         combined_split_assignment.to_csv(
-            os.path.join(self.processed_dir_main, self.splits_file_name),
-            index=False,
+            os.path.join(self.processed_dir_main, "splits.csv"), index=False
         )
 
         # Store the splits in class variables
@@ -1271,13 +1270,3 @@ def processed_file_names_dict(self) -> dict:
         if self.n_token_limit is not None:
             return {"data": f"data_maxlen{self.n_token_limit}.pt"}
         return {"data": "data.pt"}
-
-    @property
-    def splits_file_name(self) -> str:
-        """
-        Returns the name of the splits file.
-
-        Returns:
-            str: The name of the splits file.
-        """
-        return "splits.csv"

chebai/preprocessing/datasets/chebi.py

@@ -660,18 +660,6 @@ def processed_file_names_dict(self) -> dict:
             }
         return {"data": f"aug_data_var{self.aug_smiles_variations}.pt"}
 
-    @property
-    def splits_file_name(self) -> str:
-        """
-        Returns the name of the splits file.
-
-        Returns:
-            str: The name of the splits file.
-        """
-        if self.augment_smiles:
-            return f"aug_splits_var{self.aug_smiles_variations}.csv"
-        return super().splits_file_name
-
 
 class JCIExtendedBase(_ChEBIDataExtractor):
     @property