Fix error

NVIDIA · Dec 18, 2024 · 2d81d60 · 2d81d60
1 parent 7679bfb
commit 2d81d60
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/main/java/com/nvidia/spark/rapids/jni/TestHostUDF.java b/src/main/java/com/nvidia/spark/rapids/jni/TestHostUDF.java
@@ -17,11 +17,11 @@
 package com.nvidia.spark.rapids.jni;
 
 import ai.rapids.cudf.NativeDepsLoader;
-import ai.rapids.cudf.Aggregation;
+import ai.rapids.cudf.HostUDFWrapper;
 
 // A new host UDF implementation must extend Aggregation.HostUDFWrapper,
 // and override the hashCode and equals methods.
-public class TestHostUDF extends Aggregation.HostUDFWrapper {
+public class TestHostUDF extends HostUDFWrapper {
   static {
     NativeDepsLoader.loadNativeDeps();
   }

diff --git a/thirdparty/cudf b/thirdparty/cudf