diff --git a/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py b/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py index 7dd9ca9dd6cfb..6ba23384922b3 100755 --- a/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py +++ b/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py @@ -36,7 +36,11 @@ def check_session_query(mod: ast.Module, file_path: str) -> bool: errors = False for node in ast.walk(mod): if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute): - if node.func.attr == "query": + if ( + node.func.attr == "query" + and isinstance(node.func.value, ast.Name) + and node.func.value.id == "session" + ): console.print(f"Deprecated query-obj found at line {node.lineno} in {file_path}.") errors = True return errors