Update docstring of full_objective

MilesCranmer · Mar 25, 2023 · 2312011 · 2312011
1 parent 9738395
commit 2312011
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -345,7 +345,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
         function eval_loss(tree, dataset::Dataset{T,L}, options)::L where {T,L}
             prediction, flag = eval_tree_array(tree, dataset.X, options)
             if !flag
-                return T(Inf)
+                return L(Inf)
             end
             return sum((prediction .- dataset.y) .^ 2) / dataset.n
         end