diff options
Diffstat (limited to 'checkers/classes.py')
| -rw-r--r-- | checkers/classes.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/checkers/classes.py b/checkers/classes.py index 7ced271db..fd7614634 100644 --- a/checkers/classes.py +++ b/checkers/classes.py @@ -353,7 +353,10 @@ a metaclass class method.'} def _check_exit(self, node): positional = sum(1 for arg in node.args.args if arg.name != 'self') - if positional != 3 and not node.args.vararg: + if positional < 3 and not node.args.vararg: + self.add_message('bad-context-manager', + node=node) + elif positional > 3: self.add_message('bad-context-manager', node=node) |