diff --git a/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py b/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py index 8c6d11262eaa7..228b0c504d0f5 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 if isinstance(node, ast.ImportFrom):