🎨 Format Python code with psf/black

DSGT-DLP · Oct 13, 2023 · 9ab1b12 · 9ab1b12
1 parent ce571e4
commit 9ab1b12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]: