Cleaning up CodeFactor issues

src/l2hmc/experiment.py

@@ -47,13 +47,11 @@ def build_lattice(self):
             if framework == 'pytorch':
                 from l2hmc.lattice.u1.pytorch.lattice import LatticeU1
                 return LatticeU1(nchains, tuple(latvolume))
-            elif framework == 'tensorflow':
+            if framework == 'tensorflow':
                 from l2hmc.lattice.u1.tensorflow.lattice import LatticeU1
                 return LatticeU1(nchains, tuple(latvolume))
-            else:
-                raise ValueError(
-                    f'Unexpected value for `framework`: {framework}'
-                )
+
+            raise ValueError(f'Unexpected value for `framework`: {framework}')
 
         if self.config.dynamics.group == 'SU3':
             if framework == 'pytorch':

src/l2hmc/lattice/su3/tensorflow/lattice.py

@@ -100,11 +100,6 @@ def _plaquette(self, x: Tensor, u: int, v: int):
         xvu = self.g.mul(x[:, v], tf.roll(x[:, u], shift=-1, axis=v + 1))
         return self.g.trace(self.g.mul(xuv, xvu, adjoint_b=True))
 
-    def _plaq(self, x: Tensor, d: int, u: int, v: int):
-        """Calculate the 1x1 square plaquette
-        Return: U[μ](x) * U[ν](x+μ) * U†[μ](x+ν) * U†[ν](x)"""
-        pass
-
     def _wilson_loops(
             self,
             x: Tensor,