@@ -61,7 +61,7 @@ def __init__(
6161
6262 # Validate dataset
6363 if not skip_validation :
64- self ._dataset = self .validate (dataset , skip_validation , limit )
64+ self ._dataset = self .validate (dataset )
6565 else :
6666 self .logger .info ("Skipping Dataset validation..." )
6767
@@ -135,24 +135,24 @@ def on_epoch_end(self):
135135 self ._dataset .remove (remove )
136136 self ._on_epoch_end_remove = []
137137
138- def validate_list_dataset (self , dataset : list , skip_validation : bool = False ) -> list :
138+ def validate_list_dataset (self , dataset : list ) -> list :
139139 """ Validate a list dataset """
140140 validated_data = [data for data in tqdm (dataset , desc = "Validating Dataset" ) if os .path .exists (data [0 ])]
141141 if not validated_data :
142142 raise FileNotFoundError ("No valid data found in dataset." )
143143
144144 return validated_data
145145
146- def validate (self , dataset : typing .Union [str , list , pd .DataFrame ], skip_validation : bool ) -> list :
146+ def validate (self , dataset : typing .Union [str , list , pd .DataFrame ]) -> typing . Union [ list , str ] :
147147 """ Validate the dataset and return the dataset """
148148
149149 if isinstance (dataset , str ):
150150 if os .path .exists (dataset ):
151151 return dataset
152152 elif isinstance (dataset , list ):
153- return self .validate_list_dataset (dataset , skip_validation )
153+ return self .validate_list_dataset (dataset )
154154 elif isinstance (dataset , pd .DataFrame ):
155- return self .validate_list_dataset (dataset .values .tolist (), skip_validation )
155+ return self .validate_list_dataset (dataset .values .tolist ())
156156 else :
157157 raise TypeError ("Dataset must be a path, list or pandas dataframe." )
158158
0 commit comments