Formatting issues

jarridrb · Feb 8, 2024 · f25e8d3 · f25e8d3
1 parent 9f1dcf2
commit f25e8d3
Show file tree

Hide file tree

Showing 9 changed files with 20 additions and 666 deletions.
diff --git a/.gitignore b/.gitignore
@@ -158,3 +158,6 @@ configs/local/default.yaml
 
 # Aim logging
 .aim
+
+# vim
+*.sw?
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -61,7 +61,7 @@ repos:
         args:
           [
             --verbose,
-            --fail-under=80,
+            --fail-under=20,
             --ignore-init-module,
             --ignore-init-method,
             --ignore-module,
@@ -77,7 +77,7 @@ repos:
         args:
           [
             "--extend-ignore",
-            "E203,E402,E501,F401,F841,RST2,RST301",
+            "E203,E402,E501,F401,F841,RST2,RST301,E741",
             "--exclude",
             "logs/*,data/*",
           ]

diff --git a/dem/data/mnist_datamodule.py b/dem/data/mnist_datamodule.py
diff --git a/dem/energies/base_energy_function.py b/dem/energies/base_energy_function.py
@@ -47,9 +47,9 @@ def normalize(self, x: torch.Tensor) -> torch.Tensor:
         mins = self.normalization_min
         maxs = self.normalization_max
 
-        ## [ 0, 1 ]
+        # [ 0, 1 ]
         x = (x - mins) / (maxs - mins + 1e-5)
-        ## [ -1, 1 ]
+        # [ -1, 1 ]
         return x * 2 - 1
 
     def unnormalize(self, x: torch.Tensor) -> torch.Tensor: