From 73abdb9fb76258c8029ba1abbf0f36555237650f Mon Sep 17 00:00:00 2001 From: zhengp0 Date: Thu, 25 Feb 2021 19:48:04 -0800 Subject: [PATCH] fix bug in include_first_basis --- src/xspline/__about__.py | 2 +- src/xspline/core.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/xspline/__about__.py b/src/xspline/__about__.py index cc4d2fa..6b8dac6 100644 --- 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.5" +__version__ = "0.0.7" __author__ = "Peng Zheng" __email__ = "zhengp@uw.edu" diff --git a/src/xspline/core.py b/src/xspline/core.py index 0f6b4e6..4189aab 100644 --- 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