Skip to content

Commit cddaace

Browse files
committed
removed redundant "skip_validation" argument for validate() in the DataProvider class as well as needed adjustemnts.
1 parent 5ce3205 commit cddaace

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

mltu/dataProvider.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)