Skip to content

Commit

Permalink
PEP 8
Browse files Browse the repository at this point in the history
  • Loading branch information
lfarv committed Dec 19, 2022
1 parent 60e5b5c commit 435ec73
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
1 change: 0 additions & 1 deletion pyat/at/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,4 +10,3 @@
from .matching import *
from .acceptance import *
from .collective import *
from .errors import *
6 changes: 4 additions & 2 deletions pyat/at/lattice/elements.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ def _broadcast(value, shape=(2,), dtype=numpy.float64):
def _nop(value):
return value


# noinspection PyUnusedLocal
def _not_allowed(value):
raise ValueError("Attribute cannot be set")
Expand Down Expand Up @@ -217,6 +218,7 @@ class Radiative(_Radiative):


class Collective(_DictLongtMotion):
# noinspection PyAbstractClass
"""Mixin class for elements representing collective effects
Derived classes will automatically set the :py:attr:`~Element.is_collective`
Expand Down Expand Up @@ -584,8 +586,8 @@ class ThinMultipole(Element):
_BUILD_ATTRIBUTES = Element._BUILD_ATTRIBUTES + ['PolynomA',
'PolynomB']
_conversions = dict(Element._conversions,
ShiftErr=_broadcast,
RotationErr=_resize)
ShiftErr=_broadcast,
RotationErr=_resize)

def __init__(self, family_name: str, poly_a, poly_b, **kwargs):
"""
Expand Down

0 comments on commit 435ec73

Please sign in to comment.