diff --git a/.github/workflows/code-quality-main.yaml b/.github/workflows/code-quality-main.yaml
index beaaa10..cefe959 100644
--- a/.github/workflows/code-quality-main.yaml
+++ b/.github/workflows/code-quality-main.yaml
@@ -17,7 +17,7 @@ jobs:
 
       - name: Set up Python
         uses: actions/setup-python@v3
-        with: 
+        with:
           python-version: "3.10"
 
       - name: Run pre-commits
diff --git a/dem/energies/base_energy_function.py b/dem/energies/base_energy_function.py
index 79c76bc..ebf51df 100644
--- a/dem/energies/base_energy_function.py
+++ b/dem/energies/base_energy_function.py
@@ -62,7 +62,9 @@ def unnormalize(self, x: torch.Tensor) -> torch.Tensor:
         x = (x + 1) / 2
         return x * (maxs - mins) + mins
 
-    def sample_test_set(self, num_points: int, normalize: bool = False, full: bool=False) -> Optional[torch.Tensor]:
+    def sample_test_set(
+        self, num_points: int, normalize: bool = False, full: bool = False
+    ) -> Optional[torch.Tensor]:
         if self.test_set is None:
             return None