            [java.util Comparator ArrayList]
            [org.apache.spark.api.java JavaSparkContext StorageLevels
             JavaRDD JavaDoubleRDD]
-           [org.apache.spark.rdd PartitionwiseSampledRDD]))
+           [org.apache.spark.rdd PartitionwiseSampledRDD]
+           [org.apache.spark HashPartitioner Partitioner]))
 
 ;; flambo makes extensive use of kryo to serialize and deserialize clojure functions
 ;; and data structures. Here we ensure that these properties are set so they are inhereted

  ([rdd n shuffle?]
   (.coalesce rdd n shuffle?)))

(defn repartition
  "Returns a new `rdd` with exactly `n` partitions."
  [rdd n]

;; happily accepted!
;;
(ns flambo.api
  (:refer-clojure :exclude [fn map reduce first count take distinct filter group-by values partition-by])
  (:require [serializable.fn :as sfn]
            [clojure.tools.logging :as log]
            [flambo.function :refer [flat-map-function
