From 393bf0d8c423133a5d327052e49f26d551267f9b Mon Sep 17 00:00:00 2001
From: Yangkang <chenyangkang24@outlook.com>
Date: Wed, 26 Nov 2025 17:48:27 -0600
Subject: [PATCH] fix #85

---
 stemflow/model/AdaSTEM.py | 9 +++++----
 stemflow/version.py       | 2 +-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/stemflow/model/AdaSTEM.py b/stemflow/model/AdaSTEM.py
index 215dc05..e6b7645 100644
--- a/stemflow/model/AdaSTEM.py
+++ b/stemflow/model/AdaSTEM.py
@@ -1070,6 +1070,11 @@ def predict_proba(
                 Return numpy.ndarray of shape (n_samples, n_ensembles)
 
         """
+        # Setup joblib_tmp_dir
+        self.joblib_tmp_dir = initiate_joblib_tmp_dir(self.lazy_loading_dir)
+        self.duckdb_config = duckdb_config(self.max_mem, self.joblib_tmp_dir)
+        
+        #
         check_X_test(X_test, self)
         check_prediction_aggregation(aggregation)
         return_by_separate_ensembles, return_std = check_prediction_return(return_by_separate_ensembles, return_std)
@@ -1077,10 +1082,6 @@ def predict_proba(
         n_jobs = check_transform_n_jobs(self, n_jobs)
         self.base_model_method = base_model_method
         self.base_model_prediction_param = base_model_prediction_param
-
-        # Setup joblib_tmp_dir
-        self.joblib_tmp_dir = initiate_joblib_tmp_dir(self.lazy_loading_dir)
-        self.duckdb_config = duckdb_config(self.max_mem, self.joblib_tmp_dir)
         
         try:
             # predict
diff --git a/stemflow/version.py b/stemflow/version.py
index 1436d8f..bf78826 100644
--- a/stemflow/version.py
+++ b/stemflow/version.py
@@ -1 +1 @@
-__version__ = "1.1.6"
+__version__ = "1.1.7"