From 201e6a6c6313d12a27086349c7bda6704f8562d9 Mon Sep 17 00:00:00 2001
From: Ashwin Srinath <shwina@users.noreply.github.com>
Date: Tue, 24 Aug 2021 13:55:32 -0400
Subject: [PATCH] Import fix

---
 python/cudf/cudf/utils/cudautils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/cudf/cudf/utils/cudautils.py b/python/cudf/cudf/utils/cudautils.py
index fb7163c52e3..727bbb1c345 100755
--- 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