chore: rename the argument for to match Python Polars

pola-rs · Mar 10, 2024 · 4fe6d65 · 4fe6d65
1 parent 06430ff
commit 4fe6d65
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/R/extendr-wrappers.R b/R/extendr-wrappers.R
@@ -178,7 +178,7 @@ RPolarsDataFrame$to_struct <- function(name) .Call(wrap__RPolarsDataFrame__to_st
 
 RPolarsDataFrame$unnest <- function(names) .Call(wrap__RPolarsDataFrame__unnest, self, names)
 
-RPolarsDataFrame$partition_by <- function(by, maintain_order, include_keys) .Call(wrap__RPolarsDataFrame__partition_by, self, by, maintain_order, include_keys)
+RPolarsDataFrame$partition_by <- function(by, maintain_order, include_key) .Call(wrap__RPolarsDataFrame__partition_by, self, by, maintain_order, include_key)
 
 RPolarsDataFrame$export_stream <- function(stream_ptr) invisible(.Call(wrap__RPolarsDataFrame__export_stream, self, stream_ptr))
 

diff --git a/src/rust/src/rdataframe/mod.rs b/src/rust/src/rdataframe/mod.rs
@@ -328,19 +328,14 @@ impl RPolarsDataFrame {
         self.lazy().unnest(names)?.collect()
     }
 
-    pub fn partition_by(
-        &self,
-        by: Robj,
-        maintain_order: Robj,
-        include_keys: Robj,
-    ) -> RResult<List> {
+    pub fn partition_by(&self, by: Robj, maintain_order: Robj, include_key: Robj) -> RResult<List> {
         let by = robj_to!(Vec, String, by)?;
         let maintain_order = robj_to!(bool, maintain_order)?;
-        let include_keys = robj_to!(bool, include_keys)?;
+        let include_key = robj_to!(bool, include_key)?;
         let out = if maintain_order {
-            self.0.clone().partition_by_stable(by, include_keys)
+            self.0.clone().partition_by_stable(by, include_key)
         } else {
-            self.0.partition_by(by, include_keys)
+            self.0.partition_by(by, include_key)
         }
         .map_err(polars_to_rpolars_err)?;