Update deeplc.py

compomics · Nov 22, 2024 · 8e171c9 · 8e171c9
1 parent a451203
commit 8e171c9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py
@@ -70,7 +70,11 @@
 from psm_utils.io.peptide_record import peprec_to_proforma
 from psm_utils.psm import PSM
 from psm_utils.psm_list import PSMList
-from tensorflow.keras.models import load_model
+
+try:
+    from tensorflow.keras.models import load_model
+except:
+    from tensorflow.python.keras.models import load_model
 from tensorflow.python.eager import context
 
 from deeplc._exceptions import CalibrationError