pdbens addcoordset fix

prody · Apr 30, 2021 · ff803d7 · ff803d7
1 parent 788617d
commit ff803d7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/prody/ensemble/pdbensemble.py b/prody/ensemble/pdbensemble.py
@@ -289,6 +289,7 @@ def addCoordset(self, coords, weights=None, label=None, **kwargs):
 
         if not n_atoms:
             self._n_atoms = n_nodes
+            n_atoms = self._n_atoms
 
         if n_nodes == n_select and self.isSelected():
             full_coords = np.repeat(self._coords[np.newaxis, :, :], n_csets, axis=0)