@@ -170,6 +170,6 @@ def from_proto(cls, user_defined_function_proto: UserDefinedFunctionProto):
             preferred_name=user_defined_function_proto.name or None,
         )
         return PythonTransformation(
-            udf=udf,
+            udf=cast(FunctionType, udf),
             udf_string=udf_string,
         )
Original file line number	Diff line number	Diff line change
`@@ -170,6 +170,6 @@ def from_proto(cls, user_defined_function_proto: UserDefinedFunctionProto):`
`170`	`170`	`preferred_name=user_defined_function_proto.name or None,`
`171`	`171`	`)`
`172`	`172`	`return PythonTransformation(`
`173`		`- udf=udf,`
	`173`	`+ udf=cast(FunctionType, udf),`
`174`	`174`	`udf_string=udf_string,`
`175`	`175`	`)`