Fix for compatibility with AllenSDK changes

scidash · Aug 6, 2018 · d859aff · d859aff
1 parent 1b89ec0
commit d859aff
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/neuronunit/aibs.py b/neuronunit/aibs.py
@@ -74,10 +74,9 @@ def get_observation(dataset_id, kind, cached=True, quiet=False):
         ct = CellTypesApi()
         cmd = ct.get_cell(dataset_id)  # Cell metadata
         if kind == 'rheobase':
-            sweep_id = cmd['ephys_features'][0]['rheobase_sweep_id']
-        sp = get_sweep_params(dataset_id, sweep_id)
-        if kind == 'rheobase':
-            value = sp['stimulus_absolute_amplitude']
+            kind = 'ef__threshold_i_long_square'
+        value = cmd[kind]
+        if kind in ['ef__threshold_i_long_square']: # A current
             value = np.round(value, 2)  # Round to nearest hundredth of a pA.
             value *= pq.pA  # Apply units.
         db[identifier] = value

diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ matplotlib>=2.0
 neo==0.5.2
 elephant==0.4.1
 igor==0.3
-allensdk>=0.14.2
+allensdk==0.14.5
 pyNeuroML>-0.3.10
 pyNN>=0.9
 execnet