back dataclass Parameters, typing in pos3d_in_bounds

OpenPTV · Apr 5, 2024 · bd6f0dd · bd6f0dd
1 parent e4870b0
commit bd6f0dd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/openptv_python/parameters.py b/openptv_python/parameters.py
@@ -8,6 +8,7 @@
 from openptv_python.constants import TR_MAX_CAMS
 
 
+@dataclass
 class Parameters:
     """Base class for all parameters with a couple of methods."""
 

diff --git a/openptv_python/track.py b/openptv_python/track.py
@@ -189,7 +189,7 @@ def predict(prev_pos, curr_pos, output):
     output[1] = 2 * curr_pos[1] - prev_pos[1]
 
 
-def pos3d_in_bounds(pos, bounds):
+def pos3d_in_bounds(pos: np.ndarray, bounds: TrackPar) -> bool:
     """Check that all components of a pos3d are in their respective bounds.
 
     taken from a track_par object.