Minor move of import statement

timsaucer · timsaucer · commit 79e3546976cc · 2026-04-04T11:51:31.000-04:00
diff --git a/python/tests/test_context.py b/python/tests/test_context.py
@@ -31,6 +31,7 @@
     Table,
     column,
     literal,
+    udf,
 )
 
 
@@ -353,7 +354,6 @@ def test_deregister_table(ctx, database):
 
 def test_deregister_udf():
     ctx = SessionContext()
-    from datafusion import udf
 
     is_null = udf(
         lambda x: x.is_null(),
