Update to new Sensor classes

openmeeg · Apr 21, 2019 · f704375 · f704375
1 parent 46ba2b3
commit f704375
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -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
@@ -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))