define typing module in udf

shuoweil · shuoweil · commit 526ab6e4945f · 2025-10-27T21:17:20.000Z
diff --git a/bigframes/blob/_functions.py b/bigframes/blob/_functions.py
@@ -85,7 +85,7 @@ def _create_udf(self):
             self._session._anon_dataset_manager.generate_unique_resource_id()
         )
 
-        func_body = inspect.getsource(self._func)
+        func_body = "import typing\n" + inspect.getsource(self._func)
         func_name = self._func.__name__
         packages = str(list(self._requirements))
 

Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ def _create_udf(self):`
`85`	`85`	`self._session._anon_dataset_manager.generate_unique_resource_id()`
`86`	`86`	`)`
`87`	`87`
`88`		`- func_body = inspect.getsource(self._func)`
	`88`	`+ func_body = "import typing\n" + inspect.getsource(self._func)`
`89`	`89`	`func_name = self._func.__name__`
`90`	`90`	`packages = str(list(self._requirements))`
`91`	`91`