modify test for pandas 2.0.0

pmlmodelling · Apr 19, 2023 · 1dc98d8 · 1dc98d8
1 parent 0ffbb2d
commit 1dc98d8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_verts_lazy.py b/tests/test_verts_lazy.py
@@ -238,6 +238,7 @@ def foo(x):
         ds.run()
         df = ds.to_dataframe()
         depths = df.reset_index().loc[:,["depth", "bnds", "depth_bnds"]].drop_duplicates().drop(columns = "bnds").groupby(["depth"]).apply(foo).drop(columns = "depth").reset_index()
+        depths.columns = ["depth", "depth_bnds"]
         df_check = (
         df
             .reset_index()