From f70437567daa40cb7a170c2165ba0c580a676ab0 Mon Sep 17 00:00:00 2001 From: eolivi Date: Sun, 14 Apr 2019 14:24:39 +0200 Subject: [PATCH] Update to new Sensor classes --- .travis.yml | 2 +- compute_leadfields.py | 8 ++++---- om_basics.py | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.travis.yml b/.travis.yml index 6183683..cb85060 100644 --- a/.travis.yml +++ b/.travis.yml @@ -6,5 +6,5 @@ install: - pip install flake8 pydocstyle script: - - flake8 . + - flake8 --ignore=W504 . - pydocstyle . diff --git a/compute_leadfields.py b/compute_leadfields.py index 3fdb9c5..8ef719b 100644 --- a/compute_leadfields.py +++ b/compute_leadfields.py @@ -31,13 +31,13 @@ dipoles = om.Matrix(dipoles_file) -meg_sensors = om.Sensors(squids_file) +meg_sensors = om.MEGSensors(squids_file) -eeg_electrodes = om.Sensors(eeg_electrodes_file) +eeg_electrodes = om.EEGSensors(eeg_electrodes_file) -eit_electrodes = om.Sensors(eit_electrodes_file, geom) +eit_electrodes = om.EITSensors(eit_electrodes_file, geom) -ecog_electrodes = om.Sensors(ecog_electrodes_file) +ecog_electrodes = om.ECoGSensors(ecog_electrodes_file) int_electrodes = om.Matrix(internal_electrodes_file) diff --git a/om_basics.py b/om_basics.py index c5337e1..7f37308 100644 --- a/om_basics.py +++ b/om_basics.py @@ -22,14 +22,14 @@ def load_headmodel(name, prefix='data'): pot_file = op.join(prefix, name, name + '.pot') geom = om.Geometry() geom.read(str(geom_file), str(cond_file)) - sensors = om.Sensors() + sensors = om.EEGSensors() sensors.load(str(patches_file)) model = {'geometry': geom, 'sensors': sensors} if op.exists(dip_file): dipoles = om.Matrix(str(dip_file)) model['dipsources'] = dipoles if op.exists(tdcs_file): - tdcs = om.Sensors(str(tdcs_file), geom) + tdcs = om.EITSensors(str(tdcs_file), geom) model['tdcssources'] = tdcs if op.exists(pot_file): pot = om.Matrix(str(pot_file))