Fix confidence values in valid_pose_dataset fixture

neuroinformatics-unit · Jan 24, 2024 · a9f7ebb · a9f7ebb
1 parent 9d78987
commit a9f7ebb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -247,7 +247,7 @@ def valid_pose_dataset(valid_tracks_array, request):
         data_vars={
             "pose_tracks": xr.DataArray(tracks_array, dims=dim_names),
             "confidence": xr.DataArray(
-                tracks_array[..., 0],
+                np.ones(tracks_array.shape[:-1]),
                 dims=dim_names[:-1],
             ),
         },