BUG: fix calling local references with keyword arguments in query

doc/source/whatsnew/v0.25.0.rst

@@ -295,7 +295,6 @@ Bug Fixes
 ~~~~~~~~~
 
 
-
 Categorical
 ^^^^^^^^^^^
 
@@ -475,6 +474,7 @@ Other
 - Removed unused C functions from vendored UltraJSON implementation (:issue:`26198`)
 - Bug in :func:`factorize` when passing an ``ExtensionArray`` with a custom ``na_sentinel`` (:issue:`25696`).
 - Allow :class:`Index` and :class:`RangeIndex` to be passed to numpy ``min`` and ``max`` functions.
+- Allow keyword arguments for callable local reference used in the ``DataFrame.query`` string
 
 
 .. _whatsnew_0.250.contributors:

pandas/core/computation/expr.py

@@ -633,9 +633,7 @@ def visit_Call(self, node, side=None, **kwargs):
                                      "'{func}'".format(func=node.func.id))
 
                 if key.arg:
-                    # TODO: bug?
-                    kwargs.append(ast.keyword(
-                        keyword.arg, self.visit(keyword.value)))  # noqa
+                    kwargs[key.arg] = self.visit(key.value).value
 
             return self.const_type(res(*new_args, **kwargs), self.env)
 

pandas/tests/computation/test_eval.py

@@ -1345,6 +1345,38 @@ def test_multi_line_expression_local_variable(self):
         assert_frame_equal(expected, df)
         assert ans is None
 
+    def test_multi_line_expression_callable_local_variable(self):
+        df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})
+
+        def local_func(a, b):
+            return b
+
+        expected = df.copy()
+        expected['c'] = expected['a'] * local_func(1, 7)
+        expected['d'] = expected['c'] + local_func(1, 7)
+        ans = df.eval("""
+        c = a * @local_func(1, 7)
+        d = c + @local_func(1, 7)
+        """, inplace=True)
+        assert_frame_equal(expected, df)
+        assert ans is None
+
+    def test_multi_line_expression_callable_local_variable_with_kwargs(self):
+        df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})
+
+        def local_func(a, b):
+            return b
+
+        expected = df.copy()
+        expected['c'] = expected['a'] * local_func(b=7, a=1)
+        expected['d'] = expected['c'] + local_func(b=7, a=1)
+        ans = df.eval("""
+        c = a * @local_func(b=7, a=1)
+        d = c + @local_func(b=7, a=1)
+        """, inplace=True)
+        assert_frame_equal(expected, df)
+        assert ans is None
+
     def test_assignment_in_query(self):
         # GH 8664
         df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})