@@ -91,19 +91,18 @@ def process(
91
91
return args , processed_kwargs
92
92
93
93
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"""
103
96
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
107
106
108
107
def process (
109
108
self , args : tuple [Any , ...], kwargs : dict [str , Any ]
0 commit comments