Skip to content

Commit 24317b6

Browse files
dario-cosciandem0
authored andcommitted
doc update
1 parent b3ffca3 commit 24317b6

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

pina/model/spline.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -148,19 +148,19 @@ def knots(self, value):
148148

149149
self._knots = value
150150

151-
def forward(self, x_):
151+
def forward(self, x):
152152
"""
153153
Forward pass of the spline model.
154154
155-
:param torch.Tensor x_: points to be evaluated.
156-
:return: the spline evaluated at x_
155+
:param torch.Tensor x: points to be evaluated.
156+
:return: the spline evaluated at x
157157
:rtype: torch.Tensor
158158
"""
159159
t = self.knots
160160
k = self.k
161161
c = self.control_points
162162

163-
basis = map(lambda i: self.basis(x_, k, i, t)[:, None], range(len(c)))
163+
basis = map(lambda i: self.basis(x, k, i, t)[:, None], range(len(c)))
164164
y = (torch.cat(list(basis), dim=1) * c).sum(axis=1)
165165

166166
return y

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
EXTRAS = {
2222
'docs': [
23-
'sphinx',
23+
'sphinx>5.0',
2424
'sphinx_rtd_theme',
2525
'sphinx_copybutton',
2626
'sphinx_design',

0 commit comments

Comments
 (0)