Merge pull request #355 from qchempku2017/main

Fixes #353
CederGroupHub · Apr 6, 2023 · c542509 · c542509
2 parents b563bd7 + 34b862b
commit c542509
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/smol/moca/ensemble.py b/smol/moca/ensemble.py
@@ -240,10 +240,9 @@ def active_sublattices(self):
     @property
     def restricted_sites(self):
         """Get indices of all restricted sites."""
-        sites = []
-        for sublattice in self.sublattices:
-            sites += sublattice.restricted_sites
-        return sites
+        return np.concatenate(
+            [sublattice.restricted_sites for sublattice in self.sublattices]
+        )
 
     @property
     def species(self):