Replace pygam with scikit-learn

Author: RobbinBouwmeester

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

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",