zframe repartition

zinggAI · Jul 16, 2024 · 6abd121 · 6abd121
1 parent cdbcb99
commit 6abd121
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 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
@@ -82,6 +82,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,scala.collection.Seq<C> partitionExprs);
+    public ZFrame<D, R, C> repartition(scala.collection.Seq<C> partitionExprs);
 
 
     public ZFrame<D, R, C> sample(boolean repartition, float 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
@@ -223,6 +223,10 @@ public ZFrame<Dataset<Row>, Row, Column> repartition(int num,scala.collection.Se
          return new SparkFrame(df.repartition(num, partitionExprs));
     }
 
+    public ZFrame<Dataset<Row>, Row, Column> repartition(scala.collection.Seq<Column> partitionExprs){
+        return new SparkFrame(df.repartition(partitionExprs));
+   }
+
     @Override
     public Column gt(String c) {
 		return gt(this,c);