Merge pull request #26 from medusabci/developers

Merge for patch update

medusa/bci/cvep_spellers.py

@@ -438,7 +438,7 @@ def fit_dataset(self, dataset, **kwargs):
         # Feature extraction and classification
         fitted_info = self.get_inst('clf_method').fit_dataset(
             dataset=data,
-            std_epoch_rejection=3.0,
+            std_epoch_rejection=None,
             show_progress_bar=True
         )
 
@@ -809,6 +809,8 @@ def __init__(self, correct_raster_latencies=False, art_rej=None, **kwargs):
         self.correct_raster_latencies = correct_raster_latencies
 
     def _assert_consistency(self, dataset: CVEPSpellerDataset):
+        # TODO: this function is not necessary. Use CVEPSpellerDataset
+        #  consistency checker instead!
         len_seqs = set()
         fs = set()
         fps_resolution = set()
@@ -937,8 +939,10 @@ def fit_dataset(self, dataset: CVEPSpellerDataset, std_epoch_rejection=3.0,
             if show_progress_bar:
                 pbar.update(1)
 
-        # Precompute nearest channels for online artifact rejection
-        sorted_dist_ch = rec_sig.channel_set.sort_nearest_channels()
+        sorted_dist_ch = None
+        if std_epoch_rejection is not None:
+            # Precompute nearest channels for online artifact rejection
+            sorted_dist_ch = rec_sig.channel_set.sort_nearest_channels()
 
         # New bar
         if show_progress_bar:

setup.py

@@ -8,7 +8,7 @@
 setup(
     name='medusa-kernel',
     packages=find_packages(),
-    version='1.2.4',
+    version='1.2.5',
     keywords=['Signal', 'Biosignal', 'EEG', 'BCI'],
     url='https://medusabci.com/',
     author='Eduardo Santamaría-Vázquez, '