fix bug

MannLabs · Sep 11, 2024 · 25dfebb · 25dfebb
1 parent e934654
commit 25dfebb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alphadia/libtransform.py b/alphadia/libtransform.py
@@ -324,11 +324,11 @@ def forward(self, input: SpecLibBase) -> SpecLibBase:
         model_mgr = ModelManager(device=device)
 
         # will load other model than default generic
-        if self.peptdeep_model_type is not None:
+        if self.peptdeep_model_type:
             logging.info(f"Loading PeptDeep models of type {self.peptdeep_model_type}")
             model_mgr.load_installed_models(self.peptdeep_model_type)
 
-        if self.peptdeep_model_path is not None and self.peptdeep_model_path != "":
+        if self.peptdeep_model_path and self.peptdeep_model_path != "":
             if not os.path.exists(self.peptdeep_model_path):
                 raise ValueError(
                     f"PeptDeep model checkpoint folder {self.peptdeep_model_path} does not exist"