Skip to content

Commit 7c26f7e

Browse files
committed
attempt fixing macos doctest
1 parent 1fbec33 commit 7c26f7e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/lightning/pytorch/utilities/model_helpers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ def wrapper(*args: Any, **kwargs: Any) -> _R_co:
129129
)
130130
return self.method(cls_type, *args, **kwargs)
131131

132+
wrapper.__func__ = self.method
132133
return wrapper
133134

134135

0 commit comments

Comments
 (0)