Add more typing_extensions for lower Python versions

src/qusi/finite_standard_light_curve_dataset.py

@@ -1,6 +1,9 @@
 from dataclasses import dataclass
 from functools import partial
-from typing import List, Callable, Any, Self
+try:
+    from typing import List, Callable, Any, Self
+except ImportError:
+    from typing_extensions import List, Callable, Any, Self
 
 import numpy as np
 from torch.utils.data import Dataset

src/qusi/finite_test_session.py

@@ -1,4 +1,7 @@
-from typing import List
+try:
+    from typing import List
+except ImportError:
+    from typing_extensions import List
 
 import torch
 from torch.nn import Module

src/qusi/infer_session.py

@@ -1,4 +1,7 @@
-from typing import List
+try:
+    from typing import List
+except ImportError:
+    from typing_extensions import List
 
 import numpy as np
 import torch

src/qusi/light_curve_transforms.py

@@ -1,4 +1,7 @@
-from typing import Tuple
+try:
+    from typing import Tuple
+except ImportError:
+    from typing_extensions import Tuple
 
 import torch
 import numpy.typing as npt

src/qusi/train_logging_configuration.py

@@ -1,5 +1,8 @@
 from dataclasses import dataclass
-from typing import Any, Dict
+try:
+    from typing import Any, Dict
+except ImportError:
+    from typing_extensions import Any, Dict
 
 
 @dataclass

src/qusi/train_session.py

@@ -1,5 +1,8 @@
 from pathlib import Path
-from typing import List, Optional
+try:
+    from typing import List, Optional
+except ImportError:
+    from typing_extensions import List, Optional
 
 import numpy as np
 import stringcase

src/qusi/wandb_liaison.py

@@ -1,4 +1,7 @@
-from typing import Any, Dict
+try:
+    from typing import Any, Dict
+except ImportError:
+    from typing_extensions import Any, Dict
 
 import wandb as wandb