minor changes

Author: aditya0by0

chebai/models/ffn.py

@@ -20,6 +20,7 @@ def __init__(
         **kwargs
     ):
         super().__init__(**kwargs)
+
         layers = []
         current_layer_input_size = input_size
         for hidden_dim in hidden_layers:

chebai/preprocessing/datasets/base.py

@@ -1,7 +1,6 @@
 import os
 import random
 from abc import ABC, abstractmethod
-from dataclasses import dataclass
 from typing import Any, Dict, Generator, List, Optional, Tuple, Union
 
 import lightning as pl

chebai/preprocessing/datasets/chebi.py

@@ -148,7 +148,6 @@ def __init__(
             # This is to get the data from respective directory related to "chebi_version_train"
             _init_kwargs = kwargs
             _init_kwargs["chebi_version"] = self.chebi_version_train
-            _init_kwargs["_skip_data_methods_on_init"] = True
             self._chebi_version_train_obj = self.__class__(
                 single_class=self.single_class,
                 **_init_kwargs,