Update features/sample to masks

deeptrack/aberrations.py

@@ -397,7 +397,7 @@ def get(
         n: int | list[int],
         m: int | list[int],
         coefficient: float | list[float],
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> np.ndarray:
         """Applies the Zernike phase aberration to the input pupil function.
 
@@ -564,7 +564,7 @@ def __init__(
         self: "Piston", 
         *args: tuple[Any, ...], 
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         """Initializes the Piston class.
 
@@ -623,7 +623,7 @@ def __init__(
         self: VerticalTilt, 
         *args: tuple[Any, ...], 
         coefficient: PropertyLike[float | list[float]] = 1, 
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         """Initializes the VerticalTilt class.
 
@@ -682,7 +682,7 @@ def __init__(
         self: HorizontalTilt,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         """Initializes the HorizontalTilt class.
 
@@ -743,7 +743,7 @@ def __init__(
         self: ObliqueAstigmatism,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         """Initializes the ObliqueAstigmatism class.
 
@@ -802,7 +802,7 @@ def __init__(
         self: Defocus,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         """Initializes the Defocus class.
 
@@ -861,7 +861,7 @@ def __init__(
         self: Astigmatism,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         """Initializes the Astigmatism class.
 
@@ -909,7 +909,7 @@ def __init__(
         self: ObliqueTrefoil,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         super().__init__(*args, n=3, m=-3, coefficient=coefficient, **kwargs)
 
@@ -930,7 +930,7 @@ def __init__(
         self: VerticalComa,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         super().__init__(*args, n=3, m=-1, coefficient=coefficient, **kwargs)
 
@@ -951,7 +951,7 @@ def __init__(
         self: HorizontalComa,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         super().__init__(*args, n=3, m=1, coefficient=coefficient, **kwargs)
 
@@ -972,7 +972,7 @@ def __init__(
         self: Trefoil,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         super().__init__(*args, n=3, m=3, coefficient=coefficient, **kwargs)
 
@@ -993,6 +993,6 @@ def __init__(
         self: SphericalAberration,
         *args: tuple[Any, ...],
         coefficient: PropertyLike[float | list[float]] = 1,
-        **kwargs: dict[str, Any],
+        **kwargs: Any,
     ) -> None:
         super().__init__(*args, n=4, m=0, coefficient=coefficient, **kwargs)

deeptrack/elementwise.py

@@ -145,8 +145,6 @@ class ElementwiseFeature(Feature):
         Returns the result of applying the function to the input array.
 
     """
-
-    __gpu_compatible__: bool = True
 
     def __init__(
         self: ElementwiseFeature,