failing matrix test for older version of polars.

childmindresearch · Jan 17, 2025 · bf710b7 · bf710b7
1 parent 1cc9a15
commit bf710b7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit/test_metrics.py b/tests/unit/test_metrics.py
@@ -260,7 +260,8 @@ def test_interpolate_data(
 ) -> None:
     """Test the acceleration data from the interpolate."""
     expected_data = pl.read_csv(actigraph_interpolation_r_version)
-    expected_acceleration = expected_data["X", "Y", "Z"].to_numpy()
+    expected_acceleration = expected_data.select(["X", "Y", "Z"]).to_numpy()
+
     test_data = readers.read_watch_data(sample_data_gt3x)
 
     interpolated_acceleration = metrics.interpolate_measure(