Advance to v0.0.7 (#10)

Fix bug in include_first_basis
ihmeuw-msca · Feb 26, 2021 · 2302f11 · 2302f11
2 parents 14c2375 + b25c94b
commit 2302f11
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/xspline/__about__.py b/src/xspline/__about__.py
@@ -7,7 +7,7 @@
 __summary__ = "xspline: Advanced spline tools"
 __uri__ = "https://github.com/zhengp0/xspline"
 
-__version__ = "0.0.6"
+__version__ = "0.0.7"
 
 __author__ = "Peng Zheng"
 __email__ = "[email protected]"

diff --git a/src/xspline/core.py b/src/xspline/core.py
@@ -651,7 +651,7 @@ def design_mat(self, x, l_extra=False, r_extra=False):
         """
         mat = np.vstack([
             self.fun(x, idx, l_extra=l_extra, r_extra=r_extra)
-            for idx in range(self.basis_start, self.num_spline_bases)
+            for idx in range(self.basis_start, self.num_spline_bases + self.basis_start)
         ]).T
         return mat
 
@@ -679,7 +679,7 @@ def design_dmat(self, x, order, l_extra=False, r_extra=False):
         """
         dmat = np.vstack([
             self.dfun(x, order, idx, l_extra=l_extra, r_extra=r_extra)
-            for idx in range(self.basis_start, self.num_spline_bases)
+            for idx in range(self.basis_start, self.num_spline_bases + self.basis_start)
         ]).T
         return dmat
 
@@ -712,7 +712,7 @@ def design_imat(self, a, x, order, l_extra=False, r_extra=False):
         """
         imat = np.vstack([
             self.ifun(a, x, order, idx, l_extra=l_extra, r_extra=r_extra)
-            for idx in range(self.basis_start, self.num_spline_bases)
+            for idx in range(self.basis_start, self.num_spline_bases + self.basis_start)
         ]).T
         return imat