File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -413,6 +413,7 @@ def test_nondeterministic_udf2(self):
413413 pydoc .render_doc (udf (lambda : random .randint (6 , 6 ), IntegerType ()))
414414 pydoc .render_doc (random_udf )
415415 pydoc .render_doc (random_udf1 )
416+ pydoc .render_doc (udf (lambda x : x ).asNondeterministic )
416417
417418 def test_nondeterministic_udf_in_aggregate (self ):
418419 from pyspark .sql .functions import udf , sum
Original file line number Diff line number Diff line change @@ -169,8 +169,8 @@ def wrapper(*args):
169169 wrapper .returnType = self .returnType
170170 wrapper .evalType = self .evalType
171171 wrapper .deterministic = self .deterministic
172- wrapper .asNondeterministic = lambda : self . asNondeterministic (). _wrapped ()
173-
172+ wrapper .asNondeterministic = functools . wraps (
173+ self . asNondeterministic )( lambda : self . asNondeterministic (). _wrapped ())
174174 return wrapper
175175
176176 def asNondeterministic (self ):
You can’t perform that action at this time.
0 commit comments