Fix issue with vector proc_time in reading depletion_summary

Previous changes made proc_time a vector, not an (N, 1) array.
The reading of this data was not updated accordingly, until now
This commit is contained in:
Andrew Johnson 2019-06-27 10:29:56 -05:00
parent 32cb3ddcf2
commit 2efc7e8b71
No known key found for this signature in database
GPG key ID: 253418E91B7F6FEB
3 changed files with 3 additions and 3 deletions

View file

@ -8,7 +8,7 @@ from sys import exit
from h5py import get_config
from unittest import Mock
from unittest.mock import Mock
from .dummy_comm import DummyCommunicator

View file

@ -359,7 +359,7 @@ class Results(object):
if "depletion time" in handle:
proc_time_dset = handle["/depletion time"]
if step < proc_time_dset.shape[0]:
results.proc_time = proc_time_dset[step, :]
results.proc_time = proc_time_dset[step]
if results.proc_time is None:
results.proc_time = np.array([np.nan])

View file

@ -116,5 +116,5 @@ class ResultsList(list):
"""
times = np.empty((len(self) - 1, 1))
for ix, res in enumerate(self[:-1]):
times[ix] = res.proc_time[0]
times[ix] = res.proc_time
return times