Replace pygam with scikit-learn

compomics · Aug 30, 2024 · 4bebd13 · 4bebd13
1 parent f3afbe0
commit 4bebd13
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py
@@ -80,8 +80,6 @@
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 # Feature extraction
-from pygam import LinearGAM, s
-
 from deeplc.feat_extractor import FeatExtractor
 
 
@@ -253,9 +251,6 @@ def __init__(
         self.pygam_calibration = pygam_calibration
         self.deeplc_retrain = deeplc_retrain
 
-        if self.pygam_calibration:
-            from pygam import LinearGAM, s
-
         self.deepcallc_mod = deepcallc_mod
 
         self.predict_ccs = predict_ccs

diff --git a/pyproject.toml b/pyproject.toml
@@ -35,7 +35,7 @@ dependencies = [
     "numpy>=1.17,<2",
     "pandas>=0.25",
     "h5py>=2.10.0,<4",
-    "pygam>=0.8.0",
+    "scikit-learn>=1.2.0",
     "deeplcretrainer>=1,<2",
     "psm_utils>=0.2.3",
     "hdf5plugin>=4.1.1",