From a673cfb1c33c35d4e151329f882a00b12a03da14 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Fri, 31 Jan 2025 12:58:02 +0200
Subject: feat: Add output type to zkvms_host_io macro

---
 guests_macro/src/lib.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'guests_macro/src')

diff --git a/guests_macro/src/lib.rs b/guests_macro/src/lib.rs
index f87e8a5..0ab7ec2 100644
--- a/guests_macro/src/lib.rs
+++ b/guests_macro/src/lib.rs
@@ -8,7 +8,8 @@ pub fn proving_entrypoint(_: TokenStream, mut item: TokenStream) -> TokenStream
 
     // Put the file in zkVMs-benchmarks/guests/
     let mut output = File::create("../type.txt").unwrap();
-    write!(output, "{}", args);
+    writeln!(output, "{}", args);
+    write!(output, "{}", ret);
 
     item.extend(format!("#[macro_export]
         macro_rules! entrypoint_expr {{
-- 
cgit v1.2.3