Yaltik/hy
2
0
Fork 0
Code Pull Requests Releases Activity

Fix Python 2 issue in take-nth

Browse Source 
It was trying to `return` a value in a generator, which Python 2 forbids.
This commit is contained in:
Kodi Arfer 2017-02-04 11:20:01 -08:00
parent 3ec919278d
commit 9f4b630e14
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
hy/core/language.hy
View File

@ -419,14 +419,13 @@
(defn take-nth [n coll] (defn take-nth [n coll]
"Return every nth member of coll "Return every nth member of coll
raises ValueError for (not (pos? n))" raises ValueError for (not (pos? n))"
(if (pos? n) (if (not (pos? n))
(do (raise (ValueError "n must be positive")))
(setv citer (iter coll) skip (dec n)) (setv citer (iter coll) skip (dec n))
(for* [val citer] (for* [val citer]
(yield val) (yield val)
(for* [_ (range skip)] (for* [_ (range skip)]
(next citer)))) (next citer))))
(raise (ValueError "n must be positive"))))
(defn zero? [n] (defn zero? [n]
"Return true if n is 0" "Return true if n is 0"