[FEAT] Feat Include Undamaged Node Classification #98
@@ -133,10 +133,7 @@ class DataProcessor:
|
||||
|
||||
# Fill the list with data
|
||||
for group_idx, file_list in self.file_index.items():
|
||||
# Adjust index to be 0-based
|
||||
list_idx = group_idx
|
||||
data[list_idx] = [self._load_dataframe(file) for file in file_list]
|
||||
|
||||
data[group_idx] = [self._load_dataframe(file) for file in file_list]
|
||||
return data
|
||||
|
||||
def get_group_data(self, group_idx: int) -> List[pd.DataFrame]:
|
||||
|
||||
Reference in New Issue
Block a user