Import fix (#9104)

Closes #9084 Authors: - Ashwin Srinath (https://github.com/shwina) Approvers: - GALI PREM SAGAR (https://github.com/galipremsagar) URL: #9104
rapidsai · Aug 25, 2021 · 359be0a · 359be0a
1 parent 5647b53
commit 359be0a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/cudf/cudf/utils/cudautils.py b/python/cudf/cudf/utils/cudautils.py
@@ -240,6 +240,7 @@ def compile_udf(udf, type_signature):
       An numpy type
 
     """
+    import cudf.core.udf
 
     # Check if we've already compiled a similar (but possibly distinct)
     # function before