type in _fit_reg

marcosdanieldasilva · Sep 15, 2024 · e675fcb · e675fcb
1 parent 240a85f
commit e675fcb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/linear-regression.jl b/src/linear-regression.jl
@@ -53,7 +53,7 @@ function _fit_regression!(fitted_models::Vector{TableRegressionModel},
       # Construct the formula
       formula = FormulaTerm(y, x)
       # Construct the ModelFrame
-      mf = ModelFrame(formula, null_schema, cols, LinearModel)
+      mf = ModelFrame(formula, null_schema, cols, type)
       # Construct the ModelMatrix
       mm = ModelMatrix(X, asgn(formula))
       # Pass the fitted_model to TableRegressionModel structure