added ruff

Author: nikolasibalic

arc/advanced/population_lifetime.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+# ruff: noqa: E741
 from scipy.integrate import odeint
 from lmfit import minimize, Parameters, report_fit
 import matplotlib.pyplot as plt

arc/alkali_atom_data.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+# ruff: noqa: E741
 """
     This module specifies properties of individual alkali metals.
 

arc/alkali_atom_functions.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+# ruff: noqa: E741
 """
 Implements general single-atom calculations
 
@@ -46,7 +47,7 @@
 import sys
 import os
 
-from typing import List, Union, Tuple, cast, Tuple, Any, Callable
+from typing import List, Union, cast, Tuple, Any, Callable
 
 if sys.version_info > (2,):
     xrange = range
@@ -3084,9 +3085,7 @@ def getLandegfExact(
             f * (f + 1) - self.I * (self.I + 1) + j * (j + 1.0)
         ) / (2 * f * (f + 1.0)) + self.gI * (
             f * (f + 1.0) + self.I * (self.I + 1.0) - j * (j + 1.0)
-        ) / (
-            2.0 * f * (f + 1.0)
-        )
+        ) / (2.0 * f * (f + 1.0))
         return gf
 
     def getHFSEnergyShift(

arc/arc_c_extensions.py

@@ -1,2 +1,2 @@
 def NumerovWavefunction(*args):
-    pass  # noqa: W292
+    pass

arc/calculations_atom_pairstate.py

@@ -1,30 +1,31 @@
 # -*- coding: utf-8 -*-
+# ruff: noqa: E741
 
 """
-    Pair-state basis level diagram calculations
-
-    Calculates Rydberg spaghetti of level diagrams, as well as pertubative C6
-    and similar properties. It also allows calculation of Foster resonances
-    tuned by DC electric fields.
-
-    Example:
-        Calculation of the Rydberg eigenstates in pair-state basis for Rubidium
-        in the vicinity of the
-        :math:`|60~S_{1/2}~m_j=1/2,~60~S_{1/2}~m_j=1/2\\rangle` state. Colour
-        highlights coupling strength from state :math:`6~P_{1/2}~m_j=1/2` with
-        :math:`\\pi` (:math:`q=0`) polarized light.
-        eigenstates::
-
-            from arc import *
-            calc1 = PairStateInteractions(Rubidium(), 60, 0, 0.5, 60, 0, 0.5,0.5, 0.5)
-            calc1.defineBasis( 0., 0., 4, 5,10e9)
-            # optionally we can save now results of calculation for future use
-            saveCalculation(calc1,"mycalculation.pkl")
-            calculation1.diagonalise(linspace(1,10.0,30),250,progressOutput = True,drivingFromState=[6,1,0.5,0.5,0])
-            calc1.plotLevelDiagram()
-            calc1.ax.set_xlim(1,10)
-            calc1.ax.set_ylim(-2,2)
-            calc1.showPlot()
+Pair-state basis level diagram calculations
+
+Calculates Rydberg spaghetti of level diagrams, as well as pertubative C6
+and similar properties. It also allows calculation of Foster resonances
+tuned by DC electric fields.
+
+Example:
+    Calculation of the Rydberg eigenstates in pair-state basis for Rubidium
+    in the vicinity of the
+    :math:`|60~S_{1/2}~m_j=1/2,~60~S_{1/2}~m_j=1/2\\rangle` state. Colour
+    highlights coupling strength from state :math:`6~P_{1/2}~m_j=1/2` with
+    :math:`\\pi` (:math:`q=0`) polarized light.
+    eigenstates::
+
+        from arc import *
+        calc1 = PairStateInteractions(Rubidium(), 60, 0, 0.5, 60, 0, 0.5,0.5, 0.5)
+        calc1.defineBasis( 0., 0., 4, 5,10e9)
+        # optionally we can save now results of calculation for future use
+        saveCalculation(calc1,"mycalculation.pkl")
+        calculation1.diagonalise(linspace(1,10.0,30),250,progressOutput = True,drivingFromState=[6,1,0.5,0.5,0])
+        calc1.plotLevelDiagram()
+        calc1.ax.set_xlim(1,10)
+        calc1.ax.set_ylim(-2,2)
+        calc1.showPlot()
 
 """
 

arc/calculations_atom_single.py

@@ -1,10 +1,10 @@
 # -*- coding: utf-8 -*-
-
+# ruff: noqa: E741
 """
-    This module provides calculations of single-atom properties.
+This module provides calculations of single-atom properties.
 
-    Included calculations are Stark maps, level plot visualisations,
-    lifetimes and radiative decays.
+Included calculations are Stark maps, level plot visualisations,
+lifetimes and radiative decays.
 
 """
 
@@ -611,9 +611,7 @@ def __init__(self, atom):
         See also:
             :obj:`eFieldList`, :obj:`highlight`, :obj:`diagonalise`
         """
-        self.highlight = (
-            []
-        )  # contribution of initial state there (overlap |<original state | given state>|^2)
+        self.highlight = []  # contribution of initial state there (overlap |<original state | given state>|^2)
         """
         `highlight[i]` is an array of values measuring highlighted feature in the
         eigenstates at electric field intensity `eFieldList[i]`. E.g. `highlight[i][j]`

arc/divalent_atom_data.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+# ruff: noqa: E741
 
 """
 Data sources

arc/divalent_atom_functions.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+# ruff: noqa: E741
 from __future__ import division, print_function, absolute_import
 from arc.alkali_atom_functions import AlkaliAtom, printStateLetter
 from arc.wigner import Wigner3j, Wigner6j

arc/web_functionality.py

@@ -1,3 +1,5 @@
+# ruff: noqa: E741
+
 from __future__ import print_function, absolute_import
 
 import sys

arc/wigner.py

@@ -246,7 +246,14 @@ def Wigner6j(j1, j2, j3, J1, J2, J3, verbose=False):
         msg = "WARNING!!\n" + msg
         msg += (
             "For the 6j-Symbol:\n ⎰%3.1f %3.1f %3.1f⎱\n ⎱%3.1f %3.1f %3.1f⎰"
-            % (j1, j2, j3, J1, J2, J3)
+            % (
+                j1,
+                j2,
+                j3,
+                J1,
+                J2,
+                J3,
+            )
         )
         if verbose:
             print(msg)
@@ -271,7 +278,14 @@ def Wigner6j(j1, j2, j3, J1, J2, J3, verbose=False):
         msg = "WARNING!!\n" + msg
         msg += (
             "For the 6j-Symbol:\n ⎰%3.1f %3.1f %3.1f⎱\n ⎱%3.1f %3.1f %3.1f⎰"
-            % (j1, j2, j3, J1, J2, J3)
+            % (
+                j1,
+                j2,
+                j3,
+                J1,
+                J2,
+                J3,
+            )
         )
         msg += "\n6j-Symbol is undefined when any triad has a non-integer sum"
         if verbose: