From 3dc7b097ee7b2c3d6636fd9444423870c1acd384 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gordon=20J=2E=20K=C3=B6hn?= <gordon@koehn.net>
Date: Tue, 15 Oct 2024 10:16:04 +0200
Subject: [PATCH] add thread tracking

---
 lollipop/cli/deconvolute.py | 9 +++++++++
 pyproject.toml              | 1 +
 2 files changed, 10 insertions(+)

diff --git a/lollipop/cli/deconvolute.py b/lollipop/cli/deconvolute.py
index b827b88..2bc796b 100755
--- a/lollipop/cli/deconvolute.py
+++ b/lollipop/cli/deconvolute.py
@@ -14,6 +14,7 @@
 import pandas as pd
 import numpy as np
 from tqdm import tqdm, trange
+from threadpoolctl import threadpool_info, ThreadpoolController
 
 import lollipop as ll
 
@@ -242,7 +243,11 @@ def _deconvolute_bootstrap(
                 confint=confint(**confint_params),
                 **weights,
             )
+
+            logging.warning(threadpool_info())
+
             t_kdec = t_kdec.deconv_all(**deconv_params)
+            
             if have_confint:
                 # with conf int
                 res = t_kdec.fitted.copy()
@@ -389,6 +394,10 @@ def deconvolute(
     tally_data,
     namefield,
 ):
+    
+    controller = ThreadpoolController()
+    logging.warning(f"Controller {controller.info()}")
+
     # load data
     yaml = ruamel.yaml.YAML(typ="rt")
     print("load data")
diff --git a/pyproject.toml b/pyproject.toml
index 20a3226..a7b4634 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,6 +23,7 @@ strictyaml = { version = ">=1.7", optional = true }
 tqdm = { version = ">=4.64", optional = true }
 click = { version = "^8.0", optional = true }
 click-option-group = { version = "^0.5", optional = true }
+threadpoolctl = "^3.5.0"
 
 [tool.poetry.extras]
 cli = [ "zstandard", "ruamel.yaml", "strictyaml", "tqdm", "click", "click-option-group" ]