Skip to content

Commit 0a5a82a

Browse files
committed
[ci skip] handle supply of wrong file extensions
1 parent a8286d8 commit 0a5a82a

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

mne/io/curry/curry.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@
1616
from ...utils import verbose, warn
1717
from ..base import BaseRaw
1818

19+
CURRY_SUFFIX_DATA = [".cdt", ".dat"]
20+
CURRY_SUFFIX_HDR = [".cdt.dpa", ".cdt.dpo", ".dap"]
21+
1922

2023
@verbose
2124
def read_raw_curry(fname, preload=False, verbose=None) -> "RawCurry":
@@ -61,7 +64,17 @@ class RawCurry(BaseRaw):
6164

6265
@verbose
6366
def __init__(self, fname, preload=False, verbose=None):
64-
fname = Path(fname)
67+
fname_in = Path(fname)
68+
fname = None
69+
if fname_in.suffix in CURRY_SUFFIX_DATA:
70+
fname = fname_in
71+
else:
72+
for data_suff in CURRY_SUFFIX_DATA:
73+
if fname_in.with_suffix(data_suff).exists():
74+
fname = fname_in.with_suffix(data_suff)
75+
break
76+
if not fname:
77+
raise FileNotFoundError("no curry data file found (.dat or .cdt)")
6578

6679
# use curry-python-reader
6780
try:
@@ -106,7 +119,7 @@ def __init__(self, fname, preload=False, verbose=None):
106119

107120
# extract other essential info not provided by curryreader
108121
fname_hdr = None
109-
for hdr_suff in [".cdt.dpa", ".cdt.dpo", ".dap"]:
122+
for hdr_suff in CURRY_SUFFIX_HDR:
110123
if fname.with_suffix(hdr_suff).exists():
111124
fname_hdr = fname.with_suffix(hdr_suff)
112125

0 commit comments

Comments
 (0)