Documentation

pyat/at/lattice/variables.py

@@ -131,9 +131,9 @@ def __init__(
             ring:       provided to an attempt to get the initial value of the
               variable
         """
-        self.name = self._setname(name)  #: Variable name
-        self.bounds = bounds  #: Variable bounds
-        self.delta = delta  #: Increment step
+        self.name: str = self._setname(name)  #: Variable name
+        self.bounds: tuple[Number, Number] = bounds  #: Variable bounds
+        self.delta: Number = delta  #: Increment step
         #: Maximum length of the history buffer. :py:obj:`None` means infinite
         self.history_length = history_length
         self._initial = np.nan

pyat/at/latticetools/observables.py

@@ -262,10 +262,10 @@ def __init__(
             Defines an Observable for the momentum compaction factor.
         """
         name = fun.__name__ if name is None else name
-        self.fun = fun
-        self.needs = needs or set()
-        self.name = name
-        self.target = target
+        self.fun: Callable = fun  #: Evaluation function
+        self.needs: set[Need] = needs or set()  #: Set of requirements
+        self.name: str = name  #: Observable name
+        self.target = target  #: Target value
         self.w = weight
         self.lbound, self.ubound = bounds
         self.initial = None
@@ -1072,13 +1072,15 @@ def ringeval(
                 r_out = ring.lattice_pass(r_in.copy(), 1, refpts=self.passrefs)
                 trajs = r_out[:, 0, :, 0].T
 
-            if not needs.isdisjoint({
-                Need.ORBIT,
-                Need.MATRIX,
-                Need.LOCALOPTICS,
-                Need.GLOBALOPTICS,
-                Need.EMITTANCE,
-            }):
+            if not needs.isdisjoint(
+                {
+                    Need.ORBIT,
+                    Need.MATRIX,
+                    Need.LOCALOPTICS,
+                    Need.GLOBALOPTICS,
+                    Need.EMITTANCE,
+                }
+            ):
                 orbit0 = self.kwargs.get("orbit", None)
                 o0, orbits = ring.find_orbit(
                     refpts=self.orbitrefs, dp=dp, dct=dct, df=df, orbit=orbit0