remove List and Tuple

Author: ggalloni

litebird_sim/detectors.py

@@ -401,10 +401,10 @@ class FreqChannelInfo:
         number_of_detectors (int): Number of detectors in the channel. If 0, inferred from
             `detector_names` or `detector_objs`. Default is 0.
 
-        detector_names (List[str]): List of detector names. If not provided, generated automatically.
+        detector_names (list[str]): List of detector names. If not provided, generated automatically.
             Default is an empty list.
 
-        detector_objs (List[UUID]): List of UUIDs (or strings convertible to UUIDs) identifying
+        detector_objs (list[UUID]): List of UUIDs (or strings convertible to UUIDs) identifying
             detector entries in the IMO. Default is an empty list.
     """
 
@@ -555,12 +555,12 @@ class InstrumentInfo:
 
         number_of_channels (int): Number of frequency channels in the instrument. Default is 0.
 
-        channel_names (List[str]): List of frequency channel names. Default is an empty list.
+        channel_names (list[str]): List of frequency channel names. Default is an empty list.
 
-        channel_objs (List[UUID]): List of references to channel definitions (UUIDs or convertible strings).
+        channel_objs (list[UUID]): List of references to channel definitions (UUIDs or convertible strings).
             Automatically converted to UUIDs. Default is an empty list.
 
-        wafer_names (List[str]): List of wafer names used in the instrument. Default is an empty list.
+        wafer_names (list[str]): List of wafer names used in the instrument. Default is an empty list.
 
         wafer_space_cm (float): Spacing between wafers, in centimeters. Default is 0.0.
 

litebird_sim/distribute.py

@@ -226,7 +226,7 @@ def distribute_optimally(elements, num_of_groups, weight_fn=None) -> list[Span]:
 
     max_weight = _partition(elements, len(elements), num_of_groups, weight_fn)
 
-    result = []  # type: List[Span]
+    result = []  # type: list[Span]
     start_idx = 0
     weight = 0
     cur_num = 0

litebird_sim/imo/imo.py

@@ -57,7 +57,7 @@ def __init__(
             else:
                 raise ValueError("You must either provide flatfile_location= or url=")
 
-        self.queried_objects = set()  # type: Set[Tuple[type, UUID]]
+        self.queried_objects = set()  # type: set[tuple[type, UUID]]
 
     def query_entity(self, identifier: UUID, track=True) -> Entity:
         """Return a :class:`.Entity` object from an UUID.

litebird_sim/io.py

@@ -470,7 +470,7 @@ def write_list_of_observations(
         # Empty list
         # We do not want to return here, as we still need to participate to
         # the call to _compute_global_start_index below
-        observations = []  # type: List[Observation]
+        observations = []  # type: list[Observation]
 
     if not isinstance(path, Path):
         path = Path(path)
@@ -695,7 +695,7 @@ def read_one_observation(
 
 
 def _build_file_entry_table(file_name_list: list[str | Path]) -> list[_FileEntry]:
-    file_entries = []  # type: List[_FileEntry]
+    file_entries = []  # type: list[_FileEntry]
     for cur_file_name in file_name_list:
         with h5py.File(cur_file_name, "r") as inpf:
             try:

litebird_sim/madam.py

@@ -297,8 +297,8 @@ def save_simulation_for_madam(
 
     # Build a dictionary containing the characteristics of each detector
     # to be written in the simulation file for Madam
-    madam_detectors = []  # type:List[Dict[str, Any]]
-    sorted_obs_per_det = []  # type: List[List[_ObsInMpiProcess]]
+    madam_detectors = []  # type:list[Dict[str, Any]]
+    sorted_obs_per_det = []  # type: list[list[_ObsInMpiProcess]]
     for det_idx, det in enumerate(detectors):
         det_id = det_idx + 1
         madam_detectors.append(

litebird_sim/mapmaking/destriper.py

@@ -1313,7 +1313,7 @@ def _run_destriper(
 
     old_r_dot = _mpi_dot(z, r)
 
-    history_of_stopping_factors = [_get_stopping_factor(r)]  # type: List[float]
+    history_of_stopping_factors = [_get_stopping_factor(r)]  # type: list[float]
     if callback:
         callback(
             stopping_factor=history_of_stopping_factors[-1],

litebird_sim/observations.py

@@ -354,7 +354,7 @@ def _make_detector_blocks(self, detectors, comm):
             and return `n_blocks_det = 3`
 
         Args:
-            detectors (List[dict]): List of detectors
+            detectors (list[dict]): List of detectors
 
             comm: The MPI communicator
 

litebird_sim/pointing_sys.py

@@ -206,14 +206,14 @@ def left_multiply_syst_quats(
 class FocalplaneCoord:
     """This class create an instance of focal plane to add offset and disturbance to the detectors.
 
-    Methods in this class multiply systematic quaternions to :attr:`.Observation.quat` (List[:class:`.RotQuaternion`]).
+    Methods in this class multiply systematic quaternions to :attr:`.Observation.quat` (list[:class:`.RotQuaternion`]).
 
     Args:
         sim (`Simulation`): :class:`.Simulation` instance.
 
         obs (`Observation`): :class:`.Observation` instance whose :attr:`.Observation.quat` is injected with the systematics.
 
-        detectors (`List[DetectorInfo]`): List of :class:`.DetectorInfo` instances.
+        detectors (`list[DetectorInfo]`): List of :class:`.DetectorInfo` instances.
     """
 
     def __init__(
@@ -228,7 +228,7 @@ def __init__(
     def add_offset(self, offset_rad, axis: str):
         """Add a rotational offset to the detectors in the focal plane by the specified axis.
 
-        This method multiplies systematic quaternions to :attr:`.Observation.quat` (List[:class:`.RotQuaternion`]).
+        This method multiplies systematic quaternions to :attr:`.Observation.quat` (list[:class:`.RotQuaternion`]).
 
         If the `offset_rad` is a scalar, it will be added to all the detectors in the focal plane.
         If the `offset_rad` is an array with same length of the list of detectors,
@@ -276,7 +276,7 @@ def add_offset(self, offset_rad, axis: str):
     def add_disturb(self, noise_rad_matrix: np.ndarray, axis: str):
         """Add a rotational disturbance to the detectors in the focal plane by the specified axis.
 
-        This method multiplies systematic quaternions to :attr:`.Observation.quat` (List[:class:`.RotQuaternion`]).
+        This method multiplies systematic quaternions to :attr:`.Observation.quat` (list[:class:`.RotQuaternion`]).
 
         If the `noise_rad_matrix` has the shape [`N`, `t`] where `N` is the number of detectors,
         `t` is the number of timestamps, the disturbance will be added to the detectors
@@ -339,7 +339,7 @@ class SpacecraftCoord:
 
         obs (`Observation`): :class:`.Observation` instance whose :attr:`.Observation.quat` is injected with the systematics.
 
-        detectors (`List[DetectorInfo]`): List of :class:`.DetectorInfo` instances.
+        detectors (`list[DetectorInfo]`): List of :class:`.DetectorInfo` instances.
     """
 
     def __init__(
@@ -398,7 +398,7 @@ class HWPCoord:
 
         obs (`Observation`): :class:`.Observation` instance whose :attr:`.Observation.quat` is injected with the systematics.
 
-        detectors (`List[DetectorInfo]`): List of :class:`.DetectorInfo` instances.
+        detectors (`list[DetectorInfo]`): List of :class:`.DetectorInfo` instances.
     """
 
     def __init__(
@@ -411,7 +411,7 @@ def __init__(
 
             sampling_rate_hz (`float`): The sampling rate of the detectors.
 
-            detectors (`List[DetectorInfo]`): List of :class:`.DetectorInfo` to which offset and disturbance are to be added.
+            detectors (`list[DetectorInfo]`): List of :class:`.DetectorInfo` to which offset and disturbance are to be added.
 
             ang_speed_radpsec (`float`): The angular speed of the spinning HWP.
 
@@ -526,7 +526,7 @@ class PointingSys:
 
         obs (`Observation`): :class:`.Observation` instance whose :attr:`.Observation.quat` is injected with the systematics.
 
-        detectors (`List[Detector]`): List of :class:`.DetectorInfo`.
+        detectors (`list[Detector]`): List of :class:`.DetectorInfo`.
     """
 
     def __init__(

litebird_sim/pointings.py

@@ -112,7 +112,7 @@ def get_pointings(
                 their data types are preserved.
 
         Returns:
-            Tuple[np.ndarray, np.ndarray | None]:
+            tuple[np.ndarray, np.ndarray | None]:
                 A pair `(pointing_buffer, hwp_buffer)`, containing the computed pointing
                 angles and (if applicable) HWP angles.
 

litebird_sim/pointings_in_obs.py

@@ -34,7 +34,7 @@ def prepare_pointings(
     model can be passed and automatically propagated to the detectors in each observation.
 
     Args:
-        observations (Observation or List[Observation]):
+        observations (Observation or list[Observation]):
             A single observation or a list of :class:`.Observation` objects to configure.
 
         instrument (InstrumentInfo):
@@ -83,7 +83,7 @@ def precompute_pointings(
     need for on-the-fly computation during scanning or map-making operations.
 
     Args:
-        observations (Observation or List[Observation]):
+        observations (Observation or list[Observation]):
             A single observation or a list of observations for which pointings should be precomputed.
 
         pointings_dtype (data-type, optional):
@@ -249,7 +249,7 @@ def _get_pointings_array(
 
     Returns
     -------
-    Tuple[np.ndarray, np.ndarray | None]
+    tuple[np.ndarray, np.ndarray | None]
         A tuple `(pointings, hwp_angle)`, where:
           - `pointings` is an array of shape (n_samples, 2) with [θ, φ].
           - `hwp_angle` is either the provided array or the one computed by the callable.
@@ -325,7 +325,7 @@ def _normalize_observations_and_pointings(
 
     Returns
     -------
-    Tuple[List[Observation], List[npt.NDArray]]
+    tuple[list[Observation], list[npt.NDArray]]
         The tuple of the list of observations and list of pointings
     """
 
@@ -396,7 +396,7 @@ def _get_pointings_and_pol_angles_det(
 
     Returns
     -------
-    Tuple[np.ndarray, np.ndarray]
+    tuple[np.ndarray, np.ndarray]
         A tuple containing the pointings and polarization angle of the detector
     """