Skip to content

Commit 63dbd12

Browse files
author
Thinh Nguyen
committed
improve validate_file logic
1 parent 08d1291 commit 63dbd12

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

element_array_ephys/readers/spikeglx.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,12 @@ def validate_file(self, file_type="ap"):
165165
file_path = self.root_dir / (self.root_name + f".{file_type}.bin")
166166
file_size = file_path.stat().st_size
167167

168-
meta_mapping = {"ap": self.apmeta, "lf": self.lfmeta}
169-
meta = meta_mapping[file_type]
168+
if file_type == "ap":
169+
meta = self.apmeta
170+
elif file_type == "lf":
171+
meta = self.lfmeta
172+
else:
173+
raise KeyError(f"Unknown file_type {file_type} - must be 'ap' or 'lf'")
170174

171175
if file_size != meta.meta["fileSizeBytes"]:
172176
raise IOError(

0 commit comments

Comments
 (0)