Even discretisation

MagriLab · Oct 13, 2023 · ad3c724 · ad3c724
1 parent 2338b25
commit ad3c724
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/kolsol/base/base_solver.py b/kolsol/base/base_solver.py
@@ -21,8 +21,8 @@ def __init__(self, nk: int, nf: int, re: float, ndim: int = 2) -> None:
         self.re = re
         self.ndim = ndim
 
-        self.nk_grid = 2 * self.nk + 1
-        self.mk_grid = 4 * self.nk + 1
+        self.nk_grid = 2 * self.nk
+        self.mk_grid = 4 * self.nk
 
     def dynamics(self, u_hat: np.ndarray) -> np.ndarray:
 
@@ -51,8 +51,8 @@ def __init__(self, nk: int, nf: int, re: float, ndim: int = 2) -> None:
         self.re = re
         self.ndim = ndim
 
-        self.nk_grid = 2 * self.nk + 1
-        self.mk_grid = 4 * self.nk + 1
+        self.nk_grid = 2 * self.nk
+        self.mk_grid = 4 * self.nk
 
     def dynamics(self, u_hat: torch.Tensor) -> torch.Tensor:
 

diff --git a/kolsol/numpy/solver.py b/kolsol/numpy/solver.py
@@ -99,7 +99,7 @@ def aap(self, f1: np.ndarray, f2: np.ndarray) -> np.ndarray:
         n_leading_dims = f1.ndim - self.ndim
         leading_dims = f1.shape[:n_leading_dims]
 
-        lb, ub = self.nk, 3 * self.nk + 1
+        lb, ub = self.nk, 3 * self.nk
         scaling = (self.mk_grid / self.nk_grid) ** self.ndim
 
         fhat = np.stack((f1, f2), axis=-1)

diff --git a/kolsol/torch/solver.py b/kolsol/torch/solver.py
@@ -128,7 +128,7 @@ def aap(self, f1: torch.Tensor, f2: torch.Tensor) -> torch.Tensor:
         n_leading_dims = f1.ndim - self.ndim
         leading_dims = f1.shape[:n_leading_dims]
 
-        lb, ub = self.nk, 3 * self.nk + 1
+        lb, ub = self.nk, 3 * self.nk
         scaling = (self.mk_grid / self.nk_grid) ** self.ndim
 
         fhat = torch.stack((f1, f2), dim=-1)