fixed check for NANs and INFs

Author: MSenden

code/python/cni_tlbx/IRM.py

@@ -259,7 +259,7 @@ def mapping(self, data, threshold = 100, mask = []):
                 v = voxel_index[m]
 
                 CS = np.matmul(tc, data[:, v]) / (mag_tc * mag_d[m])
-                idx_remove = (CS == np.Inf)| (CS == np.NaN);
+                idx_remove = (np.isinf(CS))| (np.isnan(CS))
                 CS[idx_remove] = 0
 
                 results['corr_fit'][v] = np.max(CS)

code/python/cni_tlbx/pRF.py

@@ -341,7 +341,7 @@ def mapping(self, data, threshold=100, mask=[]):
                 v = voxel_index[m]
 
                 CS = np.matmul(tc, data[:, m]) / (mag_tc * mag_d[m])
-                idx_remove = (CS == np.Inf) | (CS == np.NaN)
+                idx_remove = idx_remove = (np.isinf(CS))| (np.isnan(CS))
                 CS[idx_remove] = 0
 
                 results['corr_fit'][v] = np.max(CS)