Avoid settign p0 to None during assignment

waltsims · Jan 23, 2025 · e7e5e8a · e7e5e8a
1 parent 36b7e9a
commit e7e5e8a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kwave/ksource.py b/kwave/ksource.py
@@ -58,8 +58,9 @@ def p0(self):
 
     @p0.setter
     def p0(self, val):
+        print(val.sum())
         # check size and contents
-        if len(val) == 0 or np.sum(val != 0) == 0:
+        if len(val) == 0:
             # if the initial pressure is empty, remove field
             self._p0 = None
         else: