From 1c73303b1cc887750fc4df5c29f4dec714c50e95 Mon Sep 17 00:00:00 2001
From: Alex <95913221+Pwhsky@users.noreply.github.com>
Date: Sat, 12 Jul 2025 18:12:14 +0200
Subject: [PATCH] added NDArray type hint

---
 deeptrack/elementwise.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/deeptrack/elementwise.py b/deeptrack/elementwise.py
index 3ef7210a..3c9ad383 100644
--- a/deeptrack/elementwise.py
+++ b/deeptrack/elementwise.py
@@ -1242,12 +1242,12 @@ def _floor_dispatch(x):
 
         Parameters
         ----------
-        x: np.ndarray or torch.Tensor
+        x: NDArray or torch.Tensor
             The input to transform.
 
         Returns
         -------
-        np.ndarray or torch.Tensor
+        NDArray or torch.Tensor
             The result after applying floor elementwise.
 
         """
@@ -1351,12 +1351,12 @@ def _ceil_dispatch(x):
 
         Parameters
         ----------
-        x: np.ndarray or torch.Tensor
+        x: NDArray or torch.Tensor
             The input to transform.
 
         Returns
         -------
-        np.ndarray or torch.Tensor
+        NDArray or torch.Tensor
             The result after applying ceil elementwise.
 
         """
@@ -1797,11 +1797,11 @@ def _imag_dispatch(x):
 
         Parameters
         ----------
-        x: np.ndarray or torch.Tensor
+        x: NDArray or torch.Tensor
 
         Returns
         -------
-        np.ndarray or torch.Tensor
+        NDArray or torch.Tensor
             Imaginary part of the input, or zero if real.
 
         """
@@ -2138,12 +2138,12 @@ def _sign_dispatch(x):
 
         Parameters
         ----------
-        x : np.ndarray or torch.Tensor
+        x : NDArray or torch.Tensor
             The input array or tensor whose elementwise signs will be computed.
 
         Returns
         -------
-        np.ndarray or torch.Tensor
+        NDArray or torch.Tensor
             The elementwise sign values of the input.
 
         """