Fix extract-datetime deprecation warning in ndsh benchmark

rapidsai · Nov 6, 2024 · 1a07a1e · 1a07a1e
1 parent ac5b3ed
commit 1a07a1e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cpp/benchmarks/ndsh/q09.cpp b/cpp/benchmarks/ndsh/q09.cpp
@@ -145,7 +145,8 @@ void run_ndsh_q9(nvbench::state& state,
 
   // Calculate the `nation`, `o_year`, and `amount` columns
   auto n_name = std::make_unique<cudf::column>(joined_table->column("n_name"));
-  auto o_year = cudf::datetime::extract_year(joined_table->column("o_orderdate"));
+  auto o_year = cudf::datetime::extract_datetime_component(
+    joined_table->column("o_orderdate"), cudf::datetime::datetime_component::YEAR);
   auto amount = calculate_amount(joined_table->column("l_discount"),
                                  joined_table->column("l_extendedprice"),
                                  joined_table->column("ps_supplycost"),