fix flake8

miykael · Aug 30, 2023 · b0ee188 · b0ee188
1 parent 0d7d595
commit b0ee188
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/atlasreader/atlasreader.py b/atlasreader/atlasreader.py
@@ -341,7 +341,7 @@ def read_atlas_peak(atlastype, coordinate, prob_thresh=5):
 
     # get atlas data
     checked_atlastype = check_atlases(atlastype)
-    if type(checked_atlastype) == list:
+    if isinstance(checked_atlastype, list):
         if not len(checked_atlastype) == 1:
             raise ValueError(
                 '\'{}\' is not a string or a single atlas. \'all\' '
@@ -407,7 +407,7 @@ def read_atlas_cluster(atlastype, cluster, affine, prob_thresh=5):
 
     # get atlas data
     checked_atlastype = check_atlases(atlastype)
-    if type(checked_atlastype) == list:
+    if isinstance(checked_atlastype, list):
         if not len(checked_atlastype) == 1:
             raise ValueError(
                 '\'{}\' is not a string or a single atlas. \'all\' '
@@ -572,8 +572,9 @@ def get_peak_data(clust_img, atlas='default', prob_thresh=5,
             segment = read_atlas_peak(atype, coord, prob_thresh)
             coord_info.append([atype.atlas, segment])
 
-        peak_info += [[peak if type(peak) != list else
-                       '; '.join(['{}% {}'.format(*e) for e in peak])
+        peak_info += [['; '.join(['{}% {}'.format(*e) for e in peak])
+                       if isinstance(peak, list)
+                       else peak
                        for (_, peak) in coord_info]]
 
     return np.column_stack([coords, peak_values, cluster_volume, peak_info])