Replace deprecating drop with drop_vars

neuroinformatics-unit · Dec 4, 2023 · 75938f8 · 75938f8
1 parent 623597b
commit 75938f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_unit/test_load_poses.py b/tests/test_unit/test_load_poses.py
@@ -118,8 +118,8 @@ def test_load_from_sleap_file_without_tracks(
         # the assigned default "individuals_0"
         assert ds_from_trackless.individuals == ["individual_0"]
         xr.testing.assert_allclose(
-            ds_from_trackless.drop("individuals"),
-            ds_from_tracked.drop("individuals"),
+            ds_from_trackless.drop_vars("individuals"),
+            ds_from_tracked.drop_vars("individuals"),
         )
 
     @pytest.mark.parametrize(