From 42f1a0b1e0ca58f0f297fd7bea80016f9097ddd1 Mon Sep 17 00:00:00 2001
From: ForceBru <ForceBru@users.noreply.github.com>
Date: Sun, 21 Aug 2022 23:12:44 +0300
Subject: [PATCH] Query current logger instead of global one

---
 src/train.jl | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/train.jl b/src/train.jl
index caba6fa..a3df666 100644
--- a/src/train.jl
+++ b/src/train.jl
@@ -100,14 +100,20 @@ function GMMk(n::Int, x::DataOrMatrix{T}; kind=:diag, nInit::Int=50, nIter::Int=
             xx = vcat(yy...)
         end
     end
-    min_level = Logging.min_enabled_level(global_logger())
-    if min_level ≤ Logging.Debug
-        loglevel = :iter
+
+    # Query info about the `current_logger` (not the global one),
+    # so that the user could choose a different logger:
+    #
+    # my_gmm = Logging.with_logger(Logging.NullLogger()) do GMM(...) end
+    min_level = Logging.min_enabled_level(Logging.current_logger())
+    loglevel = if min_level ≤ Logging.Debug
+        :iter
     elseif min_level ≤ Logging.Info
-        loglevel = :final
+        :final
     else
-        loglevel = :none
+        :none
     end
+
     km = Clustering.kmeans(xx'[:,:], n, maxiter=nInit, display = loglevel)
     μ::Matrix{T} = km.centers'
     if kind == :diag