precommit fixes

jarridrb · Feb 8, 2024 · 342ad12 · 342ad12
1 parent ab77405
commit 342ad12
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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