Skip to content
Snippets Groups Projects

[LPD] [Correct] Ignore missing sources when finding memory cell patterns

Merged Thomas Kluyver requested to merge fix/lpd-mem-cells-missing-mod into master
1 file
+ 6
2
Compare changes
  • Side-by-side
  • Inline
+ 6
2
@@ -4,7 +4,7 @@ from typing import Optional
from warnings import warn
import numpy as np
from extra_data import RunDirectory
from extra_data import RunDirectory, SourceNameError
def get_mem_cell_pattern(run, sources) -> np.ndarray:
@@ -15,7 +15,11 @@ def get_mem_cell_pattern(run, sources) -> np.ndarray:
"""
patterns = []
for source in sources:
cell_id_data = run[source, 'image.cellId'].drop_empty_trains()
try:
cell_id_data = run[source, 'image.cellId'].drop_empty_trains()
except SourceNameError:
continue # Source missing
if len(cell_id_data.train_ids) == 0:
continue # No data for this module
cell_ids = cell_id_data[0].ndarray().flatten()
Loading