From deab88a5d60d54c4aa15dad0471dbbc1ee02bef3 Mon Sep 17 00:00:00 2001
From: Ziyang Hu <hu.ziyang@cantab.net>
Date: Mon, 19 Jun 2023 17:15:25 +0800
Subject: [PATCH] change %return syntax

---
 cozo-core/src/cozoscript.pest | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cozo-core/src/cozoscript.pest b/cozo-core/src/cozoscript.pest
index 47313184..4e9cdd20 100644
--- a/cozo-core/src/cozoscript.pest
+++ b/cozo-core/src/cozoscript.pest
@@ -252,7 +252,7 @@ imperative_block = {imperative_stmt+}
 break_stmt = {"%break" ~ ident?}
 ignore_error_script = {"%ignore_error" ~ imperative_clause}
 continue_stmt = {"%continue" ~ ident?}
-return_stmt = {"%return" ~ (ident | underscore_ident | imperative_clause)*}
+return_stmt = {"%return" ~ (((ident | underscore_ident | imperative_clause) ~ ",")* ~ (ident | underscore_ident | imperative_clause))?}
 loop_block = {("%mark" ~ ident)? ~ "%loop" ~ imperative_block ~ "%end"}
 temp_swap = {"%swap" ~ underscore_ident ~ underscore_ident}
 debug_stmt = {"%debug" ~ (ident | underscore_ident)}