Integrator.integrate() error raise TypeError("Can't broadcast

Hi all

I’m having a go at mesh based R2S shut down dose rate and keep coming across this error when I call integrator.integrate(). Has anyone seen this one before.

[openmc.deplete] t=0.0 s, dt=1 s, source=1e+18
[openmc.deplete] t=1.0 s, dt=1 s, source=0.0
[openmc.deplete] t=2.0 s, dt=1 s, source=1e+21
[openmc.deplete] t=3.0 (final operator evaluation)
Traceback (most recent call last):
  File "3_mesh_based_shut_down_dose_rate_example.py", line 207, in <module>
    integrator.integrate(
  File "/home/jshimwell/openmc/openmc/deplete/abc.py", line 850, in integrate
    StepResult.save(self.operator, [n], res_list, [t, t],
  File "/home/jshimwell/openmc/openmc/deplete/stepresult.py", line 570, in save
    results.export_to_hdf5(path, step_ind)
  File "/home/jshimwell/openmc/openmc/deplete/stepresult.py", line 271, in export_to_hdf5
    res._to_hdf5(handle, step, parallel=False)
  File "/home/jshimwell/openmc/openmc/deplete/stepresult.py", line 426, in _to_hdf5
    number_dset[index, i, low:high+1] = self.data[i]
  File "h5py/_objects.pyx", line 54, in h5py._objects.with_phil.wrapper
  File "h5py/_objects.pyx", line 55, in h5py._objects.with_phil.wrapper
  File "/home/jshimwell/miniconda3/envs/openmc_dev/lib/python3.8/site-packages/h5py/_hl/dataset.py", line 949, in __setitem__
    mspace = h5s.create_simple(selection.expand_shape(mshape))
  File "/home/jshimwell/miniconda3/envs/openmc_dev/lib/python3.8/site-packages/h5py/_hl/selections.py", line 264, in expand_shape
    raise TypeError("Can't broadcast %s -> %s" % (source_shape, self.array_shape))  # array shape
TypeError: Can't broadcast (72, 149) -> (8, 149)
>>> 

I have attached the code in case it is useful but I will reduce this to a minimal not working example

3_mesh_based_shut_down_dose_rate_example.py (12.2 KB)

The part that confuses me is that I’m passing in equal length arrays into the operator so I’m not sure how it arrives at the different length broadcast error. We have array length checking when the operator is formed so I’m not sure what is going on