linting

BoothGroup · Dec 3, 2024 · 08a170e · 08a170e
1 parent dc46b35
commit 08a170e
Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/ebcc/ext/fci.py b/ebcc/ext/fci.py
@@ -237,7 +237,7 @@ def _generator(order: int) -> Iterator[tuple[str, NDArray[float64]]]:
 
     # FIXME: This is some representational issue to ensure that tCC is correct -- why is it 0.25
     #        and not 0.5, and should we generalise with util.symmetrise?
-    amps.t2.aaaa = (amps.t2.aaaa - amps.t2.aaaa.swapaxes(0, 1)) * 0.25
-    amps.t2.bbbb = (amps.t2.bbbb - amps.t2.bbbb.swapaxes(0, 1)) * 0.25
+    amps.t2.aaaa = (amps.t2.aaaa - amps.t2.aaaa.swapaxes(0, 1)) * 0.25  # type: ignore
+    amps.t2.bbbb = (amps.t2.bbbb - amps.t2.bbbb.swapaxes(0, 1)) * 0.25  # type: ignore
 
     return amps
diff --git a/ebcc/ext/tcc.py b/ebcc/ext/tcc.py
@@ -214,7 +214,9 @@ def _set_active_space(
             if name not in amps_active:
                 continue
             for comb in amps[name].keys():
-                masks = tuple(_mask(self.cc.space["ab".index(s)], k.upper()) for s, k in zip(comb, key))
+                masks = tuple(
+                    _mask(self.cc.space["ab".index(s)], k.upper()) for s, k in zip(comb, key)
+                )
                 indices = np.ix_(*masks)
                 _put(amps[name][comb], indices, amps_active[name][comb])  # type: ignore