make mypy happy with numpy typing

neuroinformatics-unit · Dec 18, 2024 · 3b24d6d · 3b24d6d
1 parent cfeaca8
commit 3b24d6d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/movement/io/nwb.py b/movement/io/nwb.py
@@ -327,10 +327,10 @@ def _ds_from_pose_estimation_series(
     position_array = np.asarray(pose_estimation_series.data)
 
     # extract confidence data (n_time,)
-    if getattr(pose_estimation_series, "confidence", None) is None:
-        confidence_array = np.full(position_array.shape[0], np.nan)
-    else:
+    if getattr(pose_estimation_series, "confidence", None) is not None:
         confidence_array = np.asarray(pose_estimation_series.confidence)
+    else:
+        confidence_array = np.full(position_array.shape[0], np.nan)
 
     # Compute fps from the time differences between timestamps
     fps = np.nanmedian(1 / np.diff(pose_estimation_series.timestamps))