numpy errstate required for serr

This commit is contained in:
Shikhar Kumar 2019-06-04 20:13:21 -04:00
parent 5e8f50dd2e
commit 0bfbd2c823

View file

@ -1855,8 +1855,9 @@ class CMFDRun(object):
kerr = abs(k_o - k_n) / k_n
# Calculate max error in source
serr = np.sqrt(np.sum(np.where(s_n > 0, ((s_n-s_o) / s_n)**2, 0))
/ len(s_n))
with np.errstate(divide='ignore', invalid='ignore'):
serr = np.sqrt(np.sum(np.where(s_n > 0, ((s_n-s_o) / s_n)**2, 0))
/ len(s_n))
# Check for convergence
iconv = kerr < self._cmfd_ktol and serr < self._stol