enh: better handling of inf values

AFM-analysis · Nov 9, 2023 · 232d51e · 232d51e
1 parent 4eefc29
commit 232d51e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nanite/rate/rater.py b/nanite/rate/rater.py
@@ -175,7 +175,7 @@ def load_training_set(cls, path=None, names=None, which_type=None,
         response = np.loadtxt(resp_path, dtype=float)
         if remove_nan:
             # Remove nan-values from training set
-            valid = ~np.isnan(np.sum(samples, axis=1))
+            valid = ~np.array(np.sum(np.isnan(samples), axis=1), dtype=bool)
             samples = samples[valid, :]
             # remove corresponding responses
             response = response[valid]