From 01886e87c39febcb9da1f059715bbf9f9fa37f91 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Tue, 18 Feb 2025 11:19:52 +0100
Subject: [PATCH] add Lambda to API reference

---
 docs/source/api/codegen.md         | 1 +
 src/pystencils/codegen/__init__.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/docs/source/api/codegen.md b/docs/source/api/codegen.md
index 8e374d4e5..c52f28f51 100644
--- a/docs/source/api/codegen.md
+++ b/docs/source/api/codegen.md
@@ -177,4 +177,5 @@ The following categories with target-specific options are exposed:
   Kernel
   GpuKernel
   Parameter
+  Lambda
 ```
diff --git a/src/pystencils/codegen/__init__.py b/src/pystencils/codegen/__init__.py
index fc1b70ca0..d06c18382 100644
--- a/src/pystencils/codegen/__init__.py
+++ b/src/pystencils/codegen/__init__.py
@@ -6,6 +6,7 @@ from .config import (
 from .parameters import Parameter
 from .kernel import Kernel, GpuKernel
 from .driver import create_kernel, get_driver
+from .functions import Lambda
 
 __all__ = [
     "Target",
@@ -14,6 +15,7 @@ __all__ = [
     "Parameter",
     "Kernel",
     "GpuKernel",
+    "Lambda",
     "create_kernel",
     "get_driver",
 ]
-- 
GitLab