diff --git a/training/training/core/dataset.py b/training/training/core/dataset.py index c26d30c0f..97e0c2976 100644 --- a/training/training/core/dataset.py +++ b/training/training/core/dataset.py @@ -180,7 +180,7 @@ def createTrainDataset(self) -> DataLoader: shuffle=self.shuffle, drop_last=True, ) - self.delete_datasets_from_directory() # Delete datasets after loading + self.delete_datasets_from_directory() # Delete datasets after loading return train_loader def createTestDataset(self) -> DataLoader: @@ -190,7 +190,7 @@ def createTestDataset(self) -> DataLoader: shuffle=self.shuffle, drop_last=True, ) - self.delete_datasets_from_directory() # Delete datasets after loading + self.delete_datasets_from_directory() # Delete datasets after loading return test_loader def getCategoryList(self) -> list[str]: