remove check for self.gmso_system where its not needed

cmelab · Aug 29, 2023 · 5f07ac8 · 5f07ac8
1 parent 1cd61eb
commit 5f07ac8
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/hoomd_organics/base/system.py b/hoomd_organics/base/system.py
@@ -137,9 +137,7 @@ def n_molecules(self):
 
     @property
     def n_particles(self):
-        if self.gmso_system:
-            return self.gmso_system.n_sites
-        return sum([mol.n_particles for mol in self.all_molecules])
+        return self.gmso_system.n_sites
 
     @property
     def mass(self):
@@ -152,8 +150,7 @@ def mass(self):
 
     @property
     def net_charge(self):
-        if self.gmso_system:
-            return sum(site.charge for site in self.gmso_system.sites)
+        return sum(site.charge for site in self.gmso_system.sites)
 
     @property
     def box(self):