simlify name checking

vanvalenlab · Jun 4, 2020 · f1aff41 · f1aff41
1 parent b4c48b8
commit f1aff41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caliban_toolbox/utils/io_utils.py b/caliban_toolbox/utils/io_utils.py
@@ -201,7 +201,7 @@ def load_npzs(crop_dir, log_data, verbose=True):
                 temp_npz = np.load(npz_path)
 
                 # determine how labels were named
-                labels_key = 'y' if 'y' in list(temp_npz.keys()) else 'annotated'
+                labels_key = 'y' if 'y' in temp_npz else 'annotated'
 
                 # last slice may be truncated, modify index
                 if slice == num_slices - 1: