Skip to content

Commit 2beb41b

Browse files
authored
Merge pull request #2 from ttngu207/chris-nwb
import the correct ephys module that has been activated
2 parents 6d7ad7c + b4ffe1d commit 2beb41b

File tree

3 files changed

+12
-2
lines changed

3 files changed

+12
-2
lines changed

element_array_ephys/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,7 @@ def get_logger(name):
1010
log = logging.getLogger(name)
1111
log.setLevel(os.getenv('LOGLEVEL', 'INFO'))
1212
return log
13+
14+
15+
# ephys_acute as default
16+
import ephys_acute as ephys
File renamed without changes.

element_array_ephys/export/nwb/nwb.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,15 @@
1313
from spikeinterface import extractors
1414
from tqdm import tqdm
1515

16-
from .. import probe, ephys
16+
from .. import probe, ephys_acute, ephys_chronic, ephys_no_curation
1717

18-
assert probe.schema.is_activated() and ephys.schema.is_activated()
18+
assert probe.schema.is_activated(), 'probe not yet activated'
19+
20+
for ephys in (ephys_acute, ephys_chronic, ephys_no_curation):
21+
if ephys.schema.is_activated():
22+
break
23+
else:
24+
raise AssertionError('ephys not yet activated')
1925

2026

2127
class DecimalEncoder(json.JSONEncoder):

0 commit comments

Comments
 (0)