added an if statement that check if ret_distr is set to false.

JuliaTrustworthyAI · Aug 19, 2024 · 94bddbc · 94bddbc
1 parent 3599c30
commit 94bddbc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/conformal_models/inductive_bayes_regression.jl b/src/conformal_models/inductive_bayes_regression.jl
@@ -22,6 +22,10 @@ function BayesRegressor(
     train_ratio::AbstractFloat=0.5,
 )
     @assert typeof(model) == LaplaceRegression "Model must be of type Laplace"
+    if model.ret_distr != false
+        @warn "model.ret_distr is not false. Setting model.ret_distr to false."
+        model.ret_distr = false
+    end
     return BayesRegressor(model, coverage, nothing, heuristic, train_ratio)
 end