Skip to content

Commit 2312011

Browse files
committed
Update docstring of full_objective
1 parent 9738395 commit 2312011

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pysr/sr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
345345
function eval_loss(tree, dataset::Dataset{T,L}, options)::L where {T,L}
346346
prediction, flag = eval_tree_array(tree, dataset.X, options)
347347
if !flag
348-
return T(Inf)
348+
return L(Inf)
349349
end
350350
return sum((prediction .- dataset.y) .^ 2) / dataset.n
351351
end

0 commit comments

Comments
 (0)