M-Labs
/
artiq
mirror of https://github.com/m-labs/artiq.git
2
0
Fork 2
Code Issues Projects Releases Wiki Activity

Correctly infer type of list(iterable).

This commit is contained in:
whitequark 2015-07-16 15:35:46 +03:00
parent 6cda67c0c6
commit 5756cfcebc
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
artiq/compiler/transforms/inferencer.py
View File

@ -534,7 +534,20 @@ class Inferencer(algorithm.Visitor):
arg, = node.args arg, = node.args
if builtins.is_iterable(arg.type): if builtins.is_iterable(arg.type):
pass def makenotes(printer, typea, typeb, loca, locb):
return [
diagnostic.Diagnostic("note",
"iterator returning elements of type {typea}",
{"typea": printer.name(typea)},
loca),
diagnostic.Diagnostic("note",
"iterator returning elements of type {typeb}",
{"typeb": printer.name(typeb)},
locb)
]
self._unify(node.type.find().params["elt"],
arg.type.find().params["elt"],
node.loc, arg.loc, makenotes=makenotes)
elif types.is_var(arg.type): elif types.is_var(arg.type):
pass # undetermined yet pass # undetermined yet
else: else: