From 19e9cd7ffc0e65ec9ef09a6d7dbdd3fbb6b7f997 Mon Sep 17 00:00:00 2001 From: Maxim Kurnikov Date: Thu, 31 Jan 2019 18:33:19 +0300 Subject: [PATCH] fix failing plugin tests --- mypy_django_plugin/plugins/migrations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mypy_django_plugin/plugins/migrations.py b/mypy_django_plugin/plugins/migrations.py index be71db9..4be98bd 100644 --- a/mypy_django_plugin/plugins/migrations.py +++ b/mypy_django_plugin/plugins/migrations.py @@ -21,7 +21,7 @@ def determine_model_cls_from_string_for_migrations(ctx: MethodContext) -> Type: if app_label is None: return ctx.default_return_type - if 'model_name' not in ctx.arg_names: + if 'model_name' not in ctx.callee_arg_names: return ctx.default_return_type model_name_expr = ctx.args[ctx.callee_arg_names.index('model_name')][0]