Merge pull request #67 from MagriLab/feature/it_cycle

it.cycle to iter
MagriLab · Mar 15, 2022 · fcd2349 · fcd2349
2 parents 901f657 + 655540f
commit fcd2349
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/kolsol/numpy/solver.py b/kolsol/numpy/solver.py
@@ -143,7 +143,7 @@ def dissip(self, u_hat: np.ndarray) -> np.ndarray:
         w_hat = self.vorticity(u_hat)
 
         # generate indices for each dimension
-        g = it.cycle(string.ascii_letters)
+        g = iter(string.ascii_letters)
         idx = ''.join(next(g) for _ in range(self.ndim))
 
         dissipation = oe.contract(f'...{idx} -> ...',  w_hat * np.conj(w_hat))

diff --git a/kolsol/torch/solver.py b/kolsol/torch/solver.py
@@ -170,7 +170,7 @@ def dissip(self, u_hat: torch.Tensor) -> torch.Tensor:
         w_hat = self.vorticity(u_hat)
 
         # generate indices for each dimension
-        g = it.cycle(string.ascii_letters)
+        g = iter(string.ascii_letters)
         idx = ''.join(next(g) for _ in range(self.ndim))
 
         dissipation = oe.contract(f'...{idx} -> ...', w_hat * torch.conj(w_hat))