Fixed typo that gave wrong npl and ntp values

MintonGroup · Nov 11, 2022 · 4bd9b6a · 4bd9b6a
1 parent d166719
commit 4bd9b6a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/swiftest/swiftest/simulation_class.py b/python/swiftest/swiftest/simulation_class.py
@@ -1864,9 +1864,9 @@ def _combine_and_fix_dsnew(self,dsnew):
         self.ds = xr.combine_by_coords([self.ds, dsnew])
 
         def get_nvals(ds):
-            if "Gmass" in dsnew:
+            if "Gmass" in ds:
                 ds['ntp'] = ds['id'].where(np.isnan(ds['Gmass'])).count(dim="id")
-                ds['npl'] = ds['id'].where(np.invert(np.isnan(ds['Gmass']))).count(dim="id") - 1
+                ds['npl'] = ds['id'].where(~(np.isnan(ds['Gmass']))).count(dim="id") - 1
             else:
                 ds['ntp'] = ds['id'].count(dim="id")
                 ds['npl'] = xr.full_like(ds['ntp'],0)