From cea91d307d9012cdfb8926892fe495e5ba8e1d10 Mon Sep 17 00:00:00 2001 From: Chichi Lalescu <chichilalescu@gmail.com> Date: Sat, 18 Nov 2017 09:34:43 -0700 Subject: [PATCH] fix infinity issue --- bfps/DNS.py | 3 +++ bfps/NavierStokes.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/bfps/DNS.py b/bfps/DNS.py index 3207ccce..9d57338d 100644 --- a/bfps/DNS.py +++ b/bfps/DNS.py @@ -236,6 +236,9 @@ class DNS(_code): ii0 = iter0 // self.parameters['niter_stat'] ii1 = iter1 // self.parameters['niter_stat'] self.statistics['kshell'] = data_file['kspace/kshell'].value + for kk in [-1, -2]: + if (self.statistics['kshell'][kk] == 0): + self.statistics['kshell'][kk] = np.nan self.statistics['kM'] = data_file['kspace/kM'].value self.statistics['dk'] = data_file['kspace/dk'].value computation_needed = True diff --git a/bfps/NavierStokes.py b/bfps/NavierStokes.py index 59fb907c..b7bf4ff8 100644 --- a/bfps/NavierStokes.py +++ b/bfps/NavierStokes.py @@ -598,6 +598,9 @@ class NavierStokes(_fluid_particle_base): ii0 = iter0 // self.parameters['niter_stat'] ii1 = iter1 // self.parameters['niter_stat'] self.statistics['kshell'] = data_file['kspace/kshell'].value + for kk in [-1, -2]: + if (self.statistics['kshell'][kk] == 0): + self.statistics['kshell'][kk] = np.nan self.statistics['kM'] = data_file['kspace/kM'].value self.statistics['dk'] = data_file['kspace/dk'].value computation_needed = True -- GitLab