Fixed typecheck errors

Fixed typecheck errors

Author: pellet

eegnb/devices/eeg.py

@@ -6,9 +6,8 @@
 """
 
 import sys
-import time
 import logging
-from time import sleep
+from time import sleep, time
 from datetime import datetime
 from multiprocessing import Process
 
@@ -27,8 +26,7 @@
     EEG_CHANNELS,
 )
 
-import socket, json, logging, struct
-from time import time
+import socket, json, struct
 
 
 logger = logging.getLogger(__name__)
@@ -152,9 +150,9 @@ def _start_muse(self, duration):
         self.recording = Process(target=record, args=(duration, self.save_fn))
         self.recording.start()
 
-        time.sleep(5)
+        sleep(5)
         self.stream_started = True
-        self.push_sample([99], timestamp=time.time())
+        self.push_sample([99], timestamp=time())
 
     def _stop_muse(self):
         pass
@@ -517,7 +515,7 @@ def _kf_push_sample(self, timestamp, marker, marker_name):
                                          'packet_sent': data_to_send})
 
     def _stop_kf(self):
-        
+
         self.kf_evnum+=1
         kf_stop_timestamp = int(time()*1e6)
 

eegnb/experiments/rest/eoec.py

@@ -105,6 +105,9 @@ def setup(self, instructions: bool = True):
 
 
     def present_stimulus(self, idx: int):
+        if self.outlet is None or self.open_sound is None or self.close_sound is None:
+            raise RuntimeError("setup() must be called before present_stimulus()")
+
         label = self.trials["parameter"].iloc[idx]  # 0 open, 1 closed
         if self.trials["timestamp"].iloc[idx] == 0:
             timestamp = time()