fix typo

cmelab · Aug 29, 2023 · 3081355 · 3081355
1 parent 4c7144a
commit 3081355
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/hoomd_organics/base/system.py b/hoomd_organics/base/system.py
@@ -150,7 +150,9 @@ def mass(self):
 
     @property
     def net_charge(self):
-        return sum(site.charge for site in self.gmso_system.sites)
+        return sum(
+            site.charge if site.charge else 0 for site in self.gmso_system.sites
+        )
 
     @property
     def box(self):
@@ -265,7 +267,12 @@ def _remove_hydrogens(self):
 
     def _scale_charges(self):
         """"""
-        charges = np.array([site.charge for site in self.gmso_system.sites])
+        charges = np.array(
+            [
+                site.charge if site.charge else 0
+                for site in self.gmso_system.sites
+            ]
+        )
         net_charge = sum(charges)
         abs_charge = sum(abs(charges))
         for site in self.gmso_system.sites: