fix: Keeps Enum order when converting to R

pola-rs · Oct 9, 2024 · 9eadb26 · 9eadb26
1 parent 61fb332
commit 9eadb26
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/rust/src/conversion_s_to_r.rs b/src/rust/src/conversion_s_to_r.rs
@@ -110,9 +110,11 @@ pub fn pl_series_to_list(
                     .set_class(["rpolars_raw_list", "list"])
                     .expect("this class label is always valid")
             }),
-            Enum(_, _) => s
-                .categorical()
-                .map(|ca| extendr_api::call!("factor", ca.iter_str().collect_robj()).unwrap()),
+            Enum(_, _) => s.categorical().map(|ca| {
+                let levels = ca.unique().unwrap().iter_str().collect_robj();
+                let values = ca.iter_str().collect_robj();
+                extendr_api::call!("factor", values, levels).unwrap()
+            }),
             Categorical(_, _) => s
                 .categorical()
                 .map(|ca| extendr_api::call!("factor", ca.iter_str().collect_robj()).unwrap()),