making 3.8 happy

cheind · Sep 22, 2023 · 0708b3f · 0708b3f
1 parent c4ecacc
commit 0708b3f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/motmetrics/tests/test_mot.py b/motmetrics/tests/test_mot.py
@@ -310,6 +310,9 @@ def test_merge_dataframes():
 
     idx = expect.index
     expect.index = expect.index.set_levels(
-        [idx.levels[0].astype(object), idx.levels[1].astype(object)]
+        [
+            idx.levels[0].astype(r.index.levels[0].dtype),
+            idx.levels[1].astype(r.index.levels[1].dtype),
+        ]
     )
     pd.testing.assert_frame_equal(r, expect)