Run black formatting on files

tardis/energy_input/util.py

@@ -72,7 +72,9 @@ def doppler_factor_3d(direction_vector, position_vector, time):
     velocity_vector = position_vector / time
     direction_vector_contiguous = np.ascontiguousarray(direction_vector)
     velocity_vector_contiguous = np.ascontiguousarray(velocity_vector)
-    return 1 - (np.dot(direction_vector_contiguous, velocity_vector_contiguous) / C_CGS)
+    return 1 - (
+        np.dot(direction_vector_contiguous, velocity_vector_contiguous) / C_CGS
+    )
 
 
 @njit(**njit_dict_no_parallel)

tardis/io/model/parse_radiation_field_configuration.py

@@ -168,7 +168,13 @@ def calculate_geometric_dilution_factor(geometry):
         The dilution factors for the inout geometry
     """
     value = (
-        1 - (geometry.r_inner[geometry.v_inner_boundary_index] ** 2 / geometry.r_middle**2).to(1).value
+        1
+        - (
+            geometry.r_inner[geometry.v_inner_boundary_index] ** 2
+            / geometry.r_middle**2
+        )
+        .to(1)
+        .value
     )
     value = np.clip(value, 0, None)
     return 0.5 * (1 - np.sqrt(value))

tardis/model/matter/composition.py

@@ -93,7 +93,7 @@ def __init__(
 
     def assemble_isotope_masses(self):
         isotope_mass_df = pd.Series(
-            index=self.isotopic_mass_fraction.index, data=-1,dtype="float64"
+            index=self.isotopic_mass_fraction.index, data=-1, dtype="float64"
         )
         for isotope_tuple in self.isotopic_mass_fraction.index:
             isotope_symbol = int("{:03d}{:03d}0000".format(*isotope_tuple))

tardis/plasma/properties/atomic.py

@@ -788,7 +788,7 @@ def exp1_times_exp(x):
             Output array.
         """
         x = np.asarray(x)
-        f = exp1(x) * np.exp(x, where=x<=500)
+        f = exp1(x) * np.exp(x, where=x <= 500)
         # Use Laurent series for large values to avoid infinite exponential
         mask = x > 500
         f[mask] = (x**-1 - x**-2 + 2 * x**-3 - 6 * x**-4)[mask]

tardis/plasma/properties/nlte_rate_equation_solver.py

@@ -554,9 +554,7 @@ def calculate_first_guess(
     """
     first_guess = pd.Series(0.0, index=rate_matrix_index)
     for atomic_number in atomic_numbers:
-        first_guess.at[(atomic_number, 1)] = number_density.loc[
-            atomic_number
-        ]
+        first_guess.at[(atomic_number, 1)] = number_density.loc[atomic_number]
     # TODO: After the first iteration, the new guess can be the old solution.
     first_guess = first_guess.values
     first_guess[-1] = electron_density