From 673e63edb77e8e40b6d01f620761d107b1df88f8 Mon Sep 17 00:00:00 2001
From: Sonal Goyal <sonalgoyal4@gmail.com>
Date: Sat, 22 Jun 2024 08:04:23 +0530
Subject: [PATCH] repartition to take multiple columns

---
 common/client/src/main/java/zingg/common/client/ZFrame.java   | 2 +-
 spark/client/src/main/java/zingg/spark/client/SparkFrame.java | 2 +-
 2 files changed, 2 insertions(+), 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
index 6ffa25a5a..11f0b0d89 100644
--- 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
index ce20ea9a5..e3d8896bf 100644
--- 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));
     }