Add temporary fix for run in lang.hy

jams2 · Nov 4, 2024 · 89fcf27 · 89fcf27
1 parent 86e0b24
commit 89fcf27
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/microkanren/lang.hy b/src/microkanren/lang.hy
@@ -33,8 +33,11 @@
   `(fresh ~lvars (&& #* goals)))
 
 (defmacro run [n lvars #* goals]
-  `(lfor state (core.take ~n ((fresh ~lvars ~@goals)(core.empty-state)))
-         (core.reify (tuple (gfor i (range ~(len lvars)) (core.Var i))) state.sub)))
+  `(lfor
+     x
+     (core.take ~n ((fresh ~lvars ~@goals (fn [s] (core.unit (core.reify ~(tuple lvars) s.sub))))
+                     (core.empty-state)))
+     x))
 
 (defmacro run* [lvars #* goals]
   (hy.macroexpand `(run -1 ~lvars ~@goals)))

diff --git a/tests/test_lang.hy b/tests/test_lang.hy
@@ -2,5 +2,5 @@
 (import microkanren.lang *)
 
 (defn test-run*-eq []
-  (assert (= (run* [q] (== q 5) )
+  (assert (= (run* [q] (== q 5))
              [#(5)])))