void (@NotNull Project project, @NotNull ProblemDescriptor descriptor) { final PsiElement problemElement = descriptor.getPsiElement(); final PsiFile psiFile = problemElement.getContainingFile(); if (psiFile instanceof PyFile) { AddImportHelper.addFromImportStatement(psiFile, (PyFromImportStatement)problemElement, AddImportHelper.ImportPriority.FUTURE, null); problemElement.delete(); } }