replace np.nan with math.isnan()

compomics · Aug 2, 2023 · 3d45fee · 3d45fee
1 parent 8350f7b
commit 3d45fee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ms2rescore/feature_generators/ms2pip.py b/ms2rescore/feature_generators/ms2pip.py
@@ -5,6 +5,7 @@
 import warnings
 from itertools import chain
 from typing import List, Optional, Union
+import math
 
 import numpy as np
 import pandas as pd
@@ -303,7 +304,7 @@ def _calculate_features_single(self, processing_result: ProcessingResult) -> Uni
         features = dict(
             zip(
                 self.feature_names,
-                [0.0 if ft is np.nan else ft for ft in feature_values],
+                [0.0 if math.isnan(ft) else ft for ft in feature_values],
             )
         )