Merge branch 'main' into configspace_update

EpistasisLab · Oct 24, 2024 · 2e4bc0f · 2e4bc0f
2 parents ebd3a25 + d8ea9ec
commit 2e4bc0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@ def calculate_version():
 
 setup(
     name='TPOT2',
-    python_requires='<3.12', #for configspace compatibility
+    python_requires='>=3.10, <3.12', #for configspace compatibility
     version=package_version,
     author='Pedro Ribeiro',
     packages=find_packages(),

diff --git a/tpot2/config/get_configspace.py b/tpot2/config/get_configspace.py
@@ -71,6 +71,7 @@
 from sklearn.svm import SVC, SVR, LinearSVR, LinearSVC
 from lightgbm import LGBMClassifier, LGBMRegressor
 import sklearn
+import sklearn.calibration as calibration
 from sklearn.naive_bayes import GaussianNB, BernoulliNB, MultinomialNB
 from sklearn.decomposition import FastICA, PCA
 from sklearn.cluster import FeatureAgglomeration