simplify symmetry handlings

otvam · Jan 7, 2025 · 1b75f82 · 1b75f82
1 parent fac7f28
commit 1b75f82
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/examples/examples_stl/transformer_air/run_matrix.py b/examples/examples_stl/transformer_air/run_matrix.py
@@ -106,10 +106,8 @@ def _show_matrix(matrix):
         {"src": "sec_src", "sink": "sec_sink"},
     ]
 
-    # definition of a list of permutations defining the symmetries
-    symmetry = [
-        [[0, 1], [1, 0]],
-    ]
+    # definition of a list of permutations with the symmetries
+    symmetry = [[0, 1], [1, 0]]
 
     # ########################################################
     # ### extract the impedance matrix

diff --git a/pypeec/utils/matrix.py b/pypeec/utils/matrix.py
@@ -256,9 +256,8 @@ def get_symmetry(terminal, symmetry):
     assert V_mat.shape == (n_winding, n_solution), "invalid solution: voltage matrix shape"
 
     # expand the solution with the symmetries
-    for sym_tmp in symmetry:
-        I_mat = _get_symmetry_expand(I_mat, sym_tmp)
-        V_mat = _get_symmetry_expand(V_mat, sym_tmp)
+    I_mat = _get_symmetry_expand(I_mat, symmetry)
+    V_mat = _get_symmetry_expand(V_mat, symmetry)
 
     # remove redundant solutions
     IV_mat = np.vstack((I_mat, V_mat))