How can I type check this function?

[Didier]

(defn drop-every-n-items

  "Takes a vector, and returns a vector with every n'th items removed."

  [vec :- Vec n :- Int] :- Vec

  (if (infix n <= 0)

    []

    (into [] (mapcat #(take (dec n) %) (partition-all n vec)))))

I'd like to properly type check this, and I just can't figure it out.

[Ambrose Bonnaire-Sergeant]

Untested, try this.

(ann drop-every-n-times (t/All [a] [(t/Vec a) Int -> (t/Vec a)]))

(defn drop-every-n-items

  "Takes a vector, and returns a vector with every n'th items removed."

  [vec n]

  (if (infix n <= 0)

    []

    (into [] (mapcat (t/fn [e :- (t/Vec a)] (take (dec n) e)) (partition-all n vec)))))