Merge branch 'bugfix/codegen-docstrings' into 'main'

Fix codegen issue with docstrings See merge request omniverse/warp!726
NVIDIA · Sep 17, 2024 · e6a4eda · e6a4eda
2 parents ee92dcd + 0585e35
commit e6a4eda
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -16,6 +16,7 @@
 - Fix array in-place op (`+=`, `-=`) adjoints to compute gradients correctly in the backwards pass.
 - `wp.sim.Model.edge_indices` now includes boundary edges
 - Allow shadowing builtin functions ([GH-308](https://github.com/NVIDIA/warp/issues/308))
+- Fix a bug in which Python docstrings would be created as local function variables in generated code.
 
 ## [1.3.3] - 2024-09-04
 

diff --git a/warp/codegen.py b/warp/codegen.py
@@ -1493,6 +1493,9 @@ def end_while(adj):
 
     def emit_FunctionDef(adj, node):
         for f in node.body:
+            # Skip variable creation for standalone constants, including docstrings
+            if isinstance(f, ast.Expr) and isinstance(f.value, ast.Constant):
+                continue
             adj.eval(f)
 
         if adj.return_var is not None and len(adj.return_var) == 1: