Logging fix

bcipy/simulator/helpers/log_utils.py

@@ -9,10 +9,7 @@ def format_alp_likelihoods(likelihoods, alp):
     return formatted
 
 
-def format_sample_rows(sample_rows: List[pd.Series]):
-    formatted_rows = []
-    for row in sample_rows:
-        new_row = row.drop(columns=['eeg'], axis=1, inplace=False)
-        formatted_rows.append(new_row.to_string(index=False, header=True))
-
-    return ", ".join(formatted_rows)
+def format_sample_rows(sample_rows: List[pd.Series]) -> str:
+    """Returns a tabular representation of the sample rows."""
+    return pd.DataFrame(sample_rows).drop(columns=['eeg']).to_string(
+        index=False, header=True)

bcipy/simulator/helpers/sampler.py

@@ -53,11 +53,12 @@ def sample(self, state: SimState) -> np.ndarray:
             if not len(filtered_data):
                 raise TaskConfigurationException(message="No eeg sample found with provided data and query")
 
-            row = filtered_data.sample(1)
+            row = filtered_data.sample(1).iloc[0]
             sample_rows.append(row)
 
         log.debug(f"EEG Samples: \n {format_sample_rows(sample_rows)}")
-        eeg_responses = [r['eeg'].to_numpy()[0] for r in sample_rows]
+
+        eeg_responses = [r['eeg'] for r in sample_rows]
         sample = self.model_input_reshaper(eeg_responses)
 
         return sample