From 8e171c9eedc2a9c2871df0bc64ff877fcf3aa517 Mon Sep 17 00:00:00 2001 From: RobbinBouwmeester Date: Fri, 22 Nov 2024 09:15:16 +0100 Subject: [PATCH] Update deeplc.py --- deeplc/deeplc.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py index 3ec5c13..e8f4397 100644 --- 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