Skip to content

Commit bc441d4

Browse files
committed
[API-Compat] Updated ForbidKeywordsDecorator
1 parent 3640209 commit bc441d4

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

python/paddle/utils/decorator_utils.py

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -91,19 +91,18 @@ def process(
9191
return args, processed_kwargs
9292

9393

94-
# *size => shape decorator
95-
class SizeArgsDecorator(DecoratorBase):
96-
"""
97-
Usage Example:
98-
99-
paddle.ones(1, dtype=paddle.float32)
100-
paddle.ones(1, 2, 3, dtype=paddle.float32)
101-
paddle.ones([1, 2, 3], dtype=paddle.float32)
102-
paddle.ones(size=[1, 2, 3], dtype=paddle.float32)
94+
class ForbidKeywordsDecorator(DecoratorBase):
95+
"""A decorator that hints users to use the correct `compat` functions, when erroneous keyword arguments are detected"""
10396

104-
paddle.ones([1, 2, 3], paddle.float32)
105-
paddle.ones(shape=[1, 2, 3], dtype=paddle.float32)
106-
"""
97+
def __init__(
98+
self, illegal_keys: list[str], func_name: str, correct_name: str
99+
) -> None:
100+
super().__init__()
101+
self.illegal_keys = (
102+
[illegal_keys] if isinstance(illegal_keys, str) else illegal_keys
103+
)
104+
self.func_name = func_name
105+
self.correct_name = correct_name
107106

108107
def process(
109108
self, args: tuple[Any, ...], kwargs: dict[str, Any]

0 commit comments

Comments
 (0)