Replaced deprecated method quantile and min

Signed-off-by: Yerzhaisang Taskali <yerzhaisang@Yerzhaisangs-MacBook-Pro.local>
opensearch-project · Oct 10, 2024 · b368561 · b368561
1 parent 5907c0a
commit b368561
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/dataframe/test_metrics_pytest.py b/tests/dataframe/test_metrics_pytest.py
@@ -501,7 +501,10 @@ def test_flights_agg_quantile(self, numeric_only):
             ["AvgTicketPrice", "FlightDelayMin", "dayOfWeek"]
         )
 
-        pd_quantile = pd_flights.agg(["quantile", "min"], numeric_only=numeric_only)
+        pd_quantile = pd_flights.agg(
+            [lambda x: x.quantile(0.5), lambda x: x.min()], numeric_only=numeric_only
+        )
+        pd_quantile.index = ["quantile", "min"]
         oml_quantile = oml_flights.agg(["quantile", "min"], numeric_only=numeric_only)
 
         assert_frame_equal(