Update parastell/utils.py

Co-authored-by: Paul Wilson <[email protected]>
svalinn · Oct 8, 2024 · eda0183 · eda0183
1 parent b1713bf
commit eda0183
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/parastell/utils.py b/parastell/utils.py
@@ -31,8 +31,7 @@ def enforce_helical_symmetry(matrix):
     Returns:
         matrix (2-D iterable of float): helically symmetric matrix.
     """
-    num_rows = matrix.shape[0]
-    num_columns = matrix.shape[1]
+    num_rows, num_columns = matrix.shape
 
     # Ensure rows represent closed loops
     for idx in range(num_rows):