fix case when a return value has ref

pthom · Nov 30, 2024 · 057ef46 · 057ef46
1 parent b56f3c8
commit 057ef46
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/litgen/internal/adapt_function_params/apply_all_adapters.py b/src/litgen/internal/adapt_function_params/apply_all_adapters.py
@@ -161,7 +161,12 @@ def _make_adapted_lambda_code_end(adapted_function: AdaptedFunction, lambda_adap
 
     # Fill auto_r_equal_or_void
     _fn_return_type = adapted_function.cpp_adapted_function.str_full_return_type()
-    auto_r_equal_or_void = "auto lambda_result = " if _fn_return_type != "void" else ""
+    _return_referenced = False
+
+    if hasattr(adapted_function.cpp_element(), "return_type"):
+        _return_referenced = '&' in adapted_function.cpp_element().return_type.modifiers
+
+    auto_r_equal_or_void = ("auto" + ("&" if _return_referenced else "") + " lambda_result = ") if _fn_return_type != "void" else ""
 
     # Fill function_or_lambda_to_call
     if adapted_function.lambda_to_call is not None: