From e7f56b760c6b200a5c5f7ae734480156b87a79cf Mon Sep 17 00:00:00 2001
From: Okon Samuel <39421418+OkonSamuel@users.noreply.github.com>
Date: Wed, 26 Jul 2023 00:55:23 +0100
Subject: [PATCH] fix UndefVarError in update method (#28)

---
 src/MLJCatBoostInterface.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/MLJCatBoostInterface.jl b/src/MLJCatBoostInterface.jl
index 6170db8..919a9d8 100644
--- a/src/MLJCatBoostInterface.jl
+++ b/src/MLJCatBoostInterface.jl
@@ -144,6 +144,7 @@ function MMI.update(mlj_model::CatBoostModels, verbosity::Integer, fitresult, ca
     if mlj_model.iterations > cache.mlj_model.iterations &&
        MMI.is_same_except(mlj_model, cache.mlj_model, :iterations)
         iterations = mlj_model.iterations - cache.mlj_model.iterations
+        verbose = verbosity <= 1 ? false : true
         new_model = model_init(mlj_model; verbose, iterations)
         new_model.fit(data_pool; init_model=fitresult)
         report = (feature_importances=feature_importance(new_model),)