|
32 | 32 | FLOAT_DTYPES = (np.float64, np.float32, np.float16) |
33 | 33 |
|
34 | 34 |
|
35 | | -def _deprecate_positional_args(f): |
| 35 | +def _deprecate_positional_args(func=None, *, version="0.25"): |
36 | 36 | """Decorator for methods that issues warnings for positional arguments. |
37 | 37 |
|
38 | 38 | Using the keyword-only argument syntax in pep 3102, arguments after the |
39 | 39 | * will issue a warning when passed as a positional argument. |
40 | 40 |
|
41 | 41 | Parameters |
42 | 42 | ---------- |
43 | | - f : callable |
| 43 | + func : callable, default=None |
44 | 44 | Function to check arguments on. |
| 45 | + version : callable, default="0.25" |
| 46 | + The version when positional arguments will result in error. |
45 | 47 | """ |
46 | | - sig = signature(f) |
47 | | - kwonly_args = [] |
48 | | - all_args = [] |
49 | | - |
50 | | - for name, param in sig.parameters.items(): |
51 | | - if param.kind == Parameter.POSITIONAL_OR_KEYWORD: |
52 | | - all_args.append(name) |
53 | | - elif param.kind == Parameter.KEYWORD_ONLY: |
54 | | - kwonly_args.append(name) |
55 | | - |
56 | | - @wraps(f) |
57 | | - def inner_f(*args, **kwargs): |
58 | | - extra_args = len(args) - len(all_args) |
59 | | - if extra_args <= 0: |
60 | | - return f(*args, **kwargs) |
61 | | - |
62 | | - # extra_args > 0 |
63 | | - args_msg = ['{}={}'.format(name, arg) |
64 | | - for name, arg in zip(kwonly_args[:extra_args], |
65 | | - args[-extra_args:])] |
66 | | - warnings.warn("Pass {} as keyword args. From version 0.25 " |
67 | | - "passing these as positional arguments will " |
68 | | - "result in an error".format(", ".join(args_msg)), |
69 | | - FutureWarning) |
70 | | - kwargs.update(zip(sig.parameters, args)) |
71 | | - return f(**kwargs) |
72 | | - return inner_f |
| 48 | + def _inner_deprecate_positional_args(f): |
| 49 | + sig = signature(f) |
| 50 | + kwonly_args = [] |
| 51 | + all_args = [] |
| 52 | + |
| 53 | + for name, param in sig.parameters.items(): |
| 54 | + if param.kind == Parameter.POSITIONAL_OR_KEYWORD: |
| 55 | + all_args.append(name) |
| 56 | + elif param.kind == Parameter.KEYWORD_ONLY: |
| 57 | + kwonly_args.append(name) |
| 58 | + |
| 59 | + @wraps(f) |
| 60 | + def inner_f(*args, **kwargs): |
| 61 | + extra_args = len(args) - len(all_args) |
| 62 | + if extra_args <= 0: |
| 63 | + return f(*args, **kwargs) |
| 64 | + |
| 65 | + # extra_args > 0 |
| 66 | + args_msg = ['{}={}'.format(name, arg) |
| 67 | + for name, arg in zip(kwonly_args[:extra_args], |
| 68 | + args[-extra_args:])] |
| 69 | + args_msg = ", ".join(args_msg) |
| 70 | + warnings.warn(f"Pass {args_msg} as keyword args. From version " |
| 71 | + f"{version} passing these as positional arguments " |
| 72 | + "will result in an error", FutureWarning) |
| 73 | + kwargs.update(zip(sig.parameters, args)) |
| 74 | + return f(**kwargs) |
| 75 | + return inner_f |
| 76 | + |
| 77 | + if func is not None: |
| 78 | + return _inner_deprecate_positional_args(func) |
| 79 | + |
| 80 | + return _inner_deprecate_positional_args |
73 | 81 |
|
74 | 82 |
|
75 | 83 | def _assert_all_finite(X, allow_nan=False, msg_dtype=None): |
|
0 commit comments