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