Skip to content

Commit 3813cdc

Browse files
committed
Fix
1 parent 837645d commit 3813cdc

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

xplt/line.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,11 @@
2828
def nominal_order(element):
2929
"""Get nominal element order (even if coefficients might be zero)"""
3030
if element.__class__.__name__ in [
31-
"Bend", "RBend", 'ThickSliceBend', 'ThickSliceRBend']: # avoid using type to support Views
31+
"Bend",
32+
"RBend",
33+
"ThickSliceBend",
34+
"ThickSliceRBend",
35+
]: # avoid using type to support Views
3236
return 0
3337
if hasattr(element, "length"):
3438
for n in range(10, -1, -1):
@@ -449,7 +453,7 @@ def ang(a):
449453
color=f"C{order}" if order >= 0 else "k",
450454
length=length or 0,
451455
label={
452-
0: "Bending magnet" if (arc or ('Bend' in TYPE[i])) else None,
456+
0: "Bending magnet" if (arc or ("Bend" in TYPE[i])) else None,
453457
1: "Quadrupole magnet",
454458
2: "Sextupole magnet",
455459
3: "Octupole magnet",

0 commit comments

Comments
 (0)