From 182a05b85113631a611169a6724162ae9247b55e Mon Sep 17 00:00:00 2001
From: mryouse
Date: Sun, 26 Jun 2022 01:31:14 +0000
Subject: add return types to builtins

---
 neb/std/functools.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'neb/std/functools.py')

diff --git a/neb/std/functools.py b/neb/std/functools.py
index facc97d..9e426b8 100644
--- a/neb/std/functools.py
+++ b/neb/std/functools.py
@@ -17,7 +17,7 @@ def interpretFilter(symbol, args, env, ns):
             out.append(arg)
     return List(out)
 
-FUNCTOOLS.register("filter", Builtin("filter", interpretFilter, [Arg("func", TypeEnum.ANY), Arg("list", TypeEnum.LIST)]))
+FUNCTOOLS.register("filter", Builtin("filter", interpretFilter, [Arg("func", TypeEnum.ANY), Arg("list", TypeEnum.LIST)], return_type=Type(":list")))
 
 def interpretMap(symbol, args, env, ns):
     func = args[0]
@@ -32,7 +32,7 @@ def interpretMap(symbol, args, env, ns):
         out.append(ev)
     return List(out)
 
-FUNCTOOLS.register("map", Builtin("map", interpretMap, [Arg("func", TypeEnum.ANY), Arg("list", TypeEnum.LIST)]))
+FUNCTOOLS.register("map", Builtin("map", interpretMap, [Arg("func", TypeEnum.ANY), Arg("list", TypeEnum.LIST)], return_type=Type(":list")))
 
 def interpretApply(symbol, args, env, ns):
     # TODO: to support lambdas, we can't assume the func is defined
-- 
cgit v1.2.3