More extensive typing

Co-authored-by: Clément Robert <cr52@protonmail.com>

Author: cphyc

yt/data_objects/time_series.py

@@ -443,11 +443,9 @@ def particle_trajectories(
     def _get_by_attribute(
         self,
         attribute: str,
-        value: Union[unyt_quantity, tuple[float, str]],
-        tolerance: Union[None, unyt_quantity, tuple[float, str]] = None,
-        side: Union[
-            Literal["nearest"], Literal["smaller"], Literal["larger"]
-        ] = "nearest",
+        value: Union[unyt_quantity, tuple[float, Union[Unit, str]]],
+        tolerance: Union[None, unyt_quantity, tuple[float, Union[Unit, str]]] = None,
+        side: Literal["nearest", "smaller", "larger"] = "nearest",
     ) -> "Dataset":
         r"""
         Get a dataset at or near to a given value.
@@ -549,12 +547,10 @@ def _get_by_attribute(
 
     def get_by_time(
         self,
-        time: Union[unyt_quantity, tuple],
-        tolerance: Union[None, unyt_quantity, tuple] = None,
-        side: Union[
-            Literal["nearest"], Literal["smaller"], Literal["larger"]
-        ] = "nearest",
-    ):
+        time: Union[unyt_quantity, tuple[float, Union[Unit, str]]],
+        tolerance: Union[None, unyt_quantity, tuple[float, Union[Unit, str]]] = None,
+        side: Literal["nearest", "smaller", "larger"] = "nearest",
+    ) -> Dataset:
         """
         Get a dataset at or near to a given time.
 
@@ -585,10 +581,8 @@ def get_by_redshift(
         self,
         redshift: float,
         tolerance: Optional[float] = None,
-        side: Union[
-            Literal["nearest"], Literal["smaller"], Literal["larger"]
-        ] = "nearest",
-    ):
+        side: Literal["nearest", "smaller", "larger"] = "nearest",
+    ) -> Dataset:
         """
         Get a dataset at or near to a given time.