Skip to content

Commit 825da3f

Browse files
chore: small changes where branch types are mentioned
Signed-off-by: jaapschoutenalliander <[email protected]>
1 parent b30f2ce commit 825da3f

File tree

1 file changed

+2
-2
lines changed
  • src/power_grid_model_ds/_core/model/grids

1 file changed

+2
-2
lines changed

src/power_grid_model_ds/_core/model/grids/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def branch_arrays(self) -> list[BranchArray]:
153153
return branch_arrays
154154

155155
def get_typed_branches(self, branch_ids: list[int] | npt.NDArray[np.int32]) -> BranchArray:
156-
"""Find a matching LineArray, LinkArray or TransformerArray for the given branch_ids
156+
"""Find a matching Branch-subtype array for the given branch_ids
157157
158158
Raises:
159159
ValueError:
@@ -174,7 +174,7 @@ def reverse_branches(self, branches: BranchArray):
174174
"""Reverse the direction of the branches."""
175175
if not branches.size:
176176
return
177-
if not isinstance(branches, (LineArray, LinkArray, TransformerArray)):
177+
if not isinstance(branches, (LineArray, LinkArray, TransformerArray, GenericBranchArray, AsymLineArray)):
178178
try:
179179
branches = self.get_typed_branches(branches.id)
180180
except ValueError:

0 commit comments

Comments
 (0)