repartition to take multiple columns

zinggAI · Jun 22, 2024 · 673e63e · 673e63e
1 parent 4399f4a
commit 673e63e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/common/client/src/main/java/zingg/common/client/ZFrame.java b/common/client/src/main/java/zingg/common/client/ZFrame.java
@@ -80,7 +80,7 @@ public interface ZFrame<D, R, C> {
 
 
     public ZFrame<D, R, C> repartition(int num);
-    public ZFrame<D, R, C> repartition(int num, C c);
+    public ZFrame<D, R, C> repartition(int num, C... c);
 
     public ZFrame<D, R, C> sample(boolean repartition, float num);
     public ZFrame<D, R, C> sample(boolean repartition, double num);

diff --git a/spark/client/src/main/java/zingg/spark/client/SparkFrame.java b/spark/client/src/main/java/zingg/spark/client/SparkFrame.java
@@ -215,7 +215,7 @@ public ZFrame<Dataset<Row>, Row, Column> repartition(int nul){
         return new SparkFrame(df.repartition(nul));
     }
 
-    public ZFrame<Dataset<Row>, Row, Column> repartition(int nul, Column c){
+    public ZFrame<Dataset<Row>, Row, Column> repartition(int nul, Column... c){
         return new SparkFrame(df.repartition(nul, c));
     }