fix syntax

Author: paganol

litebird_sim/mapmaking/pair_differencing.py

@@ -141,10 +141,14 @@ def _accumulate_pair_differenced_samples_and_build_nobs_matrix(
                 info_pix[1, 0] += pair_cos * pair_sin * inv_sigma2
                 info_pix[1, 1] += pair_sin * pair_sin * inv_sigma2
 
-
-    for cur_sample_t, cur_sample_b, cur_pix_idx, cur_psi_t, cur_psi_b, cur_t_mask in zip(
-        tod_t, tod_b, pix_t, psi_t, psi_b, t_mask
-    ):
+    for (
+        cur_sample_t,
+        cur_sample_b,
+        cur_pix_idx,
+        cur_psi_t,
+        cur_psi_b,
+        cur_t_mask,
+    ) in zip(tod_t, tod_b, pix_t, psi_t, psi_b, t_mask):
         if cur_t_mask:
             pair_sample = cur_sample_t - cur_sample_b
             pair_cos = np.cos(2 * cur_psi_t) - np.cos(2 * cur_psi_b)
@@ -198,7 +202,9 @@ def _build_nobs_matrix(
         )
 
         for t_idx, b_idx in detector_pairs:
-            if not np.array_equal(pixidx_all[t_idx, cur_t_mask], pixidx_all[b_idx, cur_t_mask]):
+            if not np.array_equal(
+                pixidx_all[t_idx, cur_t_mask], pixidx_all[b_idx, cur_t_mask]
+            ):
                 raise ValueError(
                     f"Observation {obs_idx}: detectors {t_idx} and {b_idx} do not "
                     "observe the same map pixels for the selected time samples."

litebird_sim/simulations.py

@@ -2216,7 +2216,9 @@ def make_pair_differenced_map(
             self.check_valid_splits(detector_split, time_split)
 
         if append_to_report and MPI_COMM_WORLD.rank == 0:
-            template_file_path = get_template_file_path("report_pair_differenced_map.md")
+            template_file_path = get_template_file_path(
+                "report_pair_differenced_map.md"
+            )
             with template_file_path.open("rt") as inpf:
                 markdown_template = "".join(inpf.readlines())
             self.append_to_report(
@@ -2264,7 +2266,9 @@ def make_pair_differenced_map_splits(
             self.check_valid_splits(detector_splits, time_splits)
 
         if append_to_report and MPI_COMM_WORLD.rank == 0:
-            template_file_path = get_template_file_path("report_pair_differenced_map_splits.md")
+            template_file_path = get_template_file_path(
+                "report_pair_differenced_map_splits.md"
+            )
             with template_file_path.open("rt") as inpf:
                 markdown_template = "".join(inpf.readlines())
             self.append_to_report(
@@ -2300,7 +2304,10 @@ def make_pair_differenced_map_splits(
 
                     filenames.append(
                         self.write_healpix_map(
-                            Path(file), mapp, column_names=names, coord=result.coordinate_system.name
+                            Path(file),
+                            mapp,
+                            column_names=names,
+                            coord=result.coordinate_system.name,
                         )
                     )
 

test/test_pair_differencing.py

@@ -7,7 +7,9 @@
 from litebird_sim.mapmaking.pair_differencing import make_pair_differenced_map
 
 
-def _build_observation(detectors: list[dict]) -> tuple[lbs.Observation, np.ndarray, int]:
+def _build_observation(
+    detectors: list[dict],
+) -> tuple[lbs.Observation, np.ndarray, int]:
     n_samples = 4
     theta, phi = hp.pix2ang(1, 0)
     psi = np.array([0.0, np.pi / 8.0, np.pi / 4.0, 3.0 * np.pi / 8.0])
@@ -71,7 +73,9 @@ def test_make_pair_differenced_map_recovers_qu():
     unseen_mask[pix_idx] = False
     assert np.all(result.binned_map[:, unseen_mask] == hp.UNSEEN)
 
-    pair_weight = 0.5 * (detectors[0]["net_ukrts"] ** 2 + detectors[1]["net_ukrts"] ** 2)
+    pair_weight = 0.5 * (
+        detectors[0]["net_ukrts"] ** 2 + detectors[1]["net_ukrts"] ** 2
+    )
     design_matrix = np.column_stack(
         (
             np.cos(2 * psi_t) - np.cos(2 * psi_b),
@@ -219,4 +223,6 @@ def test_simulation_make_pair_differenced_map_splits(tmp_path):
     )
 
     assert "full_full" in results
-    assert np.allclose(results["full_full"].binned_map[:, pix_idx], np.array([q_true, u_true]))
+    assert np.allclose(
+        results["full_full"].binned_map[:, pix_idx], np.array([q_true, u_true])
+    )