diff --git a/superset/extensions/pylint.py b/superset/extensions/pylint.py index deb0c865721..fc177884da2 100644 --- a/superset/extensions/pylint.py +++ b/superset/extensions/pylint.py @@ -87,9 +87,12 @@ class SQLParsingLibraryImportChecker(BaseChecker): def _is_disallowed(self, file_path: Path, root_mod: str) -> bool: # True if sqlglot is imported outside superset/sql, # or if any forbidden library is imported anywhere - in_superset_sql = file_path.match("**/superset/sql/**/*.py") or file_path.match( - "**/superset/sql/*.py" - ) + allowed = { + "**/supersql/sql/**/*.py", + "**/supersql/sql/*.py", + "**/superset/config.py", + } + in_superset_sql = any(file_path.match(pattern) for pattern in allowed) return (root_mod == "sqlglot" and not in_superset_sql) or root_mod in { "sqlparse", "sqloxide",