Georg Willer commited on
Commit
b9f7415
·
1 Parent(s): e76f24f

Change default header names to upercase starting & add checking for alternative time_header names

Browse files
eyetrack2saccade.py CHANGED
@@ -26,7 +26,7 @@ class Eye2SacExtractor:
26
  def _map_relevant_data(self, time_header: str, x_headers: Union[str, list], y_headers: Union[str, list]):
27
  # map and extract relevant data
28
  try:
29
- self.time = self.data[time_header].to_numpy().flatten()
30
  if isinstance(x_headers, str):
31
  self.x = self._get_value_array(x_headers, X_ALIASES)
32
  # if the user accidently specified the header as list with a single entry, capture it
 
26
  def _map_relevant_data(self, time_header: str, x_headers: Union[str, list], y_headers: Union[str, list]):
27
  # map and extract relevant data
28
  try:
29
+ self.time = self._get_value_array(time_header, TIME_ALIASES)
30
  if isinstance(x_headers, str):
31
  self.x = self._get_value_array(x_headers, X_ALIASES)
32
  # if the user accidently specified the header as list with a single entry, capture it
eyetrack_2_saccade_pipeline.py CHANGED
@@ -23,7 +23,7 @@ class Eye2SacPipeline(Pipeline):
23
  preprocess_kwargs["y_headers"] = kwargs["y_headers"]
24
  return preprocess_kwargs, {}, {}
25
 
26
- def preprocess(self, inputs, missing=0.0, minlen=5, maxvel=40, maxacc=30, time_header = 'time', x_headers = 'x', y_headers= 'y'):
27
  return self.eye2SacExtractor.extract_features(inputs, time_header, x_headers, y_headers, missing, minlen, maxvel, maxacc)
28
 
29
  def _forward(self, model_inputs):
 
23
  preprocess_kwargs["y_headers"] = kwargs["y_headers"]
24
  return preprocess_kwargs, {}, {}
25
 
26
+ def preprocess(self, inputs, missing=0.0, minlen=5, maxvel=40, maxacc=30, time_header = 'Time', x_headers = 'X', y_headers= 'Y'):
27
  return self.eye2SacExtractor.extract_features(inputs, time_header, x_headers, y_headers, missing, minlen, maxvel, maxacc)
28
 
29
  def _forward(self, model_inputs):