Better error logging

kymata-atlas · Jan 15, 2024 · 8e311fc · 8e311fc
1 parent 77906f2
commit 8e311fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/invokers/run_gridsearch.py b/invokers/run_gridsearch.py
@@ -1,8 +1,6 @@
 from pathlib import Path
 import argparse
 
-from matplotlib import pyplot
-
 from kymata.gridsearch.plain import do_gridsearch
 from kymata.io.functions import load_function
 from kymata.io.mne import load_emeg_pack

diff --git a/kymata/entities/expression.py b/kymata/entities/expression.py
@@ -92,8 +92,8 @@ def __init__(self,
                 data = data[i]
                 data = self._init_prep_data(data)
                 # Check validity of input data dimensions
-                assert len(channels) == data.shape[0], f"{channel_coord_name} mismatch for {f}"
-                assert len(latencies) == data.shape[1], f"Latencies mismatch for {f}"
+                assert len(channels) == data.shape[0], f"{channel_coord_name} mismatch for {f}: {len(channels)} {channel_coord_name} versus data shape {data.shape}"
+                assert len(latencies) == data.shape[1], f"Latencies mismatch for {f}: {len(latencies)} latencies versus data shape {data.shape}"
                 dataset_dict[layer] = DataArray(
                     data=data,
                     dims=self._dims,