quick formatting

Author: fsoubelet

pyhdtoolkit/cpymadtools/lhc/__init__.py

@@ -21,14 +21,14 @@
         .. tab-item:: Importing from the lhc module
 
             .. code-block:: python
-        
+
                 from pyhdtoolkit.cpymadtools.lhc import LHCSetup
                 # use this now
 
         .. tab-item:: Importing from the private module
 
             .. code-block:: python
-        
+
                 from pyhdtoolkit.cpymadtools.lhc._setup import LHCSetup
                 # use this now
 """

pyhdtoolkit/utils/__init__.py

@@ -1,3 +1,3 @@
-from . import cmdline, decorators, contexts, htc_monitor, logging  # noqa: TID252
+from . import cmdline, contexts, decorators, htc_monitor, logging  # noqa: TID252
 
 __all__ = ["cmdline", "decorators", "contexts", "htc_monitor", "logging"]

pyhdtoolkit/utils/_misc.py

@@ -177,7 +177,7 @@ def add_noise_to_arc_bpms(
         The BPM names should be in the index of the dataframe.
         Selection is case insensitive.
 
-        
+
     Parameters
     ----------
     df : pandas.DataFrame

pyhdtoolkit/utils/contexts.py

@@ -12,15 +12,15 @@
 import time
 
 from contextlib import contextmanager
-
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from collections.abc import Callable
     from types import NoneType
 
+
 @contextmanager
-def timeit(function: Callable) -> NoneType: # type: ignore
+def timeit(function: Callable) -> NoneType:  # type: ignore
     """
     .. versionadded:: 0.4.0
 

pyhdtoolkit/utils/decorators.py

@@ -6,11 +6,14 @@
 
 Provides useful decorators.
 """
+
 from __future__ import annotations
+
 import functools
 import inspect
 import traceback
 import warnings
+
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -32,12 +35,12 @@ def deprecated(message: str = "") -> Callable:
         Extra information to be displayed after the deprecation
         notice, when the function is used. Defaults to an empty
         string (no extra information).
-    
+
     Returns
     -------
     Callable
         The decorated function.
-    
+
     Example
     -------
         .. code-block:: python
@@ -71,6 +74,7 @@ def function_wrapper(*args, **kwargs):
 
 # ----- Utility JIT Compilation decorator ----- #
 
+
 def maybe_jit(func: Callable, **kwargs) -> Callable:
     """
     .. versionadded:: 1.7.0

pyhdtoolkit/utils/htc_monitor.py

@@ -176,7 +176,7 @@ def _make_tasks_table(tasks: list[HTCTaskSummary]) -> Table:
     tasks : list[HTCTaskSummary]
         A list of `~.models.htc.HTCTaskSummary` models, as
         parsed from the output of the ``condor_q`` command.
-    
+
     Returns
     -------
     rich.table.Table
@@ -247,7 +247,7 @@ def _process_scheduler_information_line(line: str) -> str:
     ----------
     line : str
         The line containing the scheduler information.
-    
+
     Returns
     -------
     str
@@ -266,7 +266,7 @@ def _process_task_summary_line(line: str) -> HTCTaskSummary:
     ----------
     line : str
         The line containing the task summary information.
-    
+
     Returns
     -------
     pyhdtoolkit.models.htc.HTCTaskSummary
@@ -352,7 +352,7 @@ def _default_cluster_table() -> Table:
     """
     Create the default structure for the Cluster
     Table, hard coded columns and no rows added.
-    
+
     Returns
     -------
     rich.table.Table

tests/test_optics.py

@@ -2,7 +2,9 @@
 
 import numpy as np
 import pytest
+
 from numpy.testing import assert_allclose
+
 from pyhdtoolkit.optics import ripken, twiss
 from pyhdtoolkit.optics.beam import Beam, compute_beam_parameters
 from pyhdtoolkit.optics.rdt import determine_rdt_line, rdt_to_order_and_type
@@ -62,7 +64,6 @@ def test_gamma_transition_raises():
         Beam(6500, 2.5e-6).gamma_transition(0)
 
 
-
 def test_beam_parameters():
     pc_gev = 19
     nemitt_x = 5e-6
@@ -83,7 +84,6 @@ def test_beam_parameters():
     assert_allclose(built.gamma_rel, 20.274666054506927)
 
 
-
 def test_beam_size(_fake_coordinates):
     assert_allclose(ripken._beam_size(_fake_coordinates), _fake_coordinates.std())  # noqa: SLF001
     assert_allclose(
@@ -102,9 +102,10 @@ def test_beam_size_raises(_fake_coordinates):
 @pytest.mark.parametrize("gemitt_x", [5e-6, 2.75e-6, 3.5e-6])
 @pytest.mark.parametrize("gemitt_y", [5e-6, 2.75e-6, 3.5e-6])
 def test_lebedev_size_floats(beta11, beta21, gemitt_x, gemitt_y):
-    assert_allclose(ripken.lebedev_beam_size(beta1_=beta11, beta2_=beta21, gemitt_x=gemitt_x, gemitt_y=gemitt_y), np.sqrt(
-        gemitt_x * beta11 + gemitt_y * beta21
-    ))
+    assert_allclose(
+        ripken.lebedev_beam_size(beta1_=beta11, beta2_=beta21, gemitt_x=gemitt_x, gemitt_y=gemitt_y),
+        np.sqrt(gemitt_x * beta11 + gemitt_y * beta21),
+    )
 
 
 def test_courant_snyder_transform():

tests/test_utils.py

@@ -17,14 +17,14 @@
 
 from pyhdtoolkit.utils import _misc, logging
 from pyhdtoolkit.utils.cmdline import CommandLine
+from pyhdtoolkit.utils.decorators import deprecated, maybe_jit
 from pyhdtoolkit.utils.htc_monitor import (
     ClusterSummary,
     HTCTaskSummary,
     _make_cluster_table,
     _make_tasks_table,
     read_condor_q,
 )
-from pyhdtoolkit.utils.decorators import deprecated, maybe_jit
 
 CURRENT_DIR = pathlib.Path(__file__).parent
 INPUTS_DIR = CURRENT_DIR / "inputs"