polars-df 0.2.5 → 0.3.0

data/ext/polars/src/series.rs

@@ -1,5 +1,4 @@
-use magnus::exception::arg_error;
-use magnus::{Error, RArray, Value};
+use magnus::{exception, Error, IntoValue, RArray, Value};
 use polars::prelude::*;
 use polars::series::IsSorted;
 use std::cell::RefCell;
@@ -30,11 +29,11 @@ impl RbSeries {
     }
 
     pub fn is_sorted_flag(&self) -> bool {
-        matches!(self.series.borrow().is_sorted(), IsSorted::Ascending)
+        matches!(self.series.borrow().is_sorted_flag(), IsSorted::Ascending)
     }
 
     pub fn is_sorted_reverse_flag(&self) -> bool {
-        matches!(self.series.borrow().is_sorted(), IsSorted::Descending)
+        matches!(self.series.borrow().is_sorted_flag(), IsSorted::Descending)
     }
 
     pub fn new_opt_bool(name: String, obj: RArray, strict: bool) -> RbResult<RbSeries> {
@@ -173,7 +172,7 @@ impl RbSeries {
     }
 
     pub fn get_idx(&self, idx: usize) -> RbResult<Value> {
-        Ok(Wrap(self.series.borrow().get(idx).map_err(RbPolarsErr::from)?).into())
+        Ok(Wrap(self.series.borrow().get(idx).map_err(RbPolarsErr::from)?).into_value())
     }
 
     pub fn bitand(&self, other: &RbSeries) -> RbResult<Self> {
@@ -216,7 +215,7 @@ impl RbSeries {
     }
 
     pub fn dtype(&self) -> Value {
-        Wrap(self.series.borrow().dtype().clone()).into()
+        Wrap(self.series.borrow().dtype().clone()).into_value()
     }
 
     pub fn inner_dtype(&self) -> Option<Value> {
@@ -224,15 +223,15 @@ impl RbSeries {
             .borrow()
             .dtype()
             .inner_dtype()
-            .map(|dt| Wrap(dt.clone()).into())
+            .map(|dt| Wrap(dt.clone()).into_value())
     }
 
     pub fn set_sorted(&self, reverse: bool) -> Self {
         let mut out = self.series.borrow().clone();
         if reverse {
-            out.set_sorted(IsSorted::Descending);
+            out.set_sorted_flag(IsSorted::Descending);
         } else {
-            out.set_sorted(IsSorted::Ascending)
+            out.set_sorted_flag(IsSorted::Ascending)
         }
         out.into()
     }
@@ -255,7 +254,7 @@ impl RbSeries {
                 .get(0)
                 .map_err(RbPolarsErr::from)?,
         )
-        .into())
+        .into_value())
     }
 
     pub fn min(&self) -> RbResult<Value> {
@@ -266,7 +265,7 @@ impl RbSeries {
                 .get(0)
                 .map_err(RbPolarsErr::from)?,
         )
-        .into())
+        .into_value())
     }
 
     pub fn sum(&self) -> RbResult<Value> {
@@ -277,7 +276,7 @@ impl RbSeries {
                 .get(0)
                 .map_err(RbPolarsErr::from)?,
         )
-        .into())
+        .into_value())
     }
 
     pub fn n_chunks(&self) -> usize {
@@ -288,7 +287,7 @@ impl RbSeries {
         let mut binding = self.series.borrow_mut();
         let res = binding.append(&other.series.borrow());
         if let Err(e) = res {
-            Err(Error::runtime_error(e.to_string()))
+            Err(Error::new(exception::runtime_error(), e.to_string()))
         } else {
             Ok(())
         }
@@ -304,7 +303,7 @@ impl RbSeries {
 
     pub fn new_from_index(&self, index: usize, length: usize) -> RbResult<Self> {
         if index >= self.series.borrow().len() {
-            Err(Error::new(arg_error(), "index is out of bounds"))
+            Err(Error::new(exception::arg_error(), "index is out of bounds"))
         } else {
             Ok(self.series.borrow().new_from_index(index, length).into())
         }
@@ -316,7 +315,10 @@ impl RbSeries {
             let series = self.series.borrow().filter(ca).unwrap();
             Ok(series.into())
         } else {
-            Err(Error::runtime_error("Expected a boolean mask".to_string()))
+            Err(Error::new(
+                exception::runtime_error(),
+                "Expected a boolean mask".to_string(),
+            ))
         }
     }
 
@@ -515,7 +517,7 @@ impl RbSeries {
         } else if let Ok(_s) = series.date() {
             let a = RArray::with_capacity(series.len());
             for v in series.iter() {
-                a.push::<Value>(Wrap(v).into()).unwrap();
+                a.push::<Value>(Wrap(v).into_value()).unwrap();
             }
             a
         } else {
@@ -546,7 +548,7 @@ impl RbSeries {
                 .get(0)
                 .unwrap_or(AnyValue::Null),
         )
-        .into())
+        .into_value())
     }
 
     pub fn clone(&self) -> Self {
@@ -756,11 +758,11 @@ impl RbSeries {
         Ok(RbSeries::new(s))
     }
 
-    pub fn to_dummies(&self) -> RbResult<RbDataFrame> {
+    pub fn to_dummies(&self, sep: Option<String>) -> RbResult<RbDataFrame> {
         let df = self
             .series
             .borrow()
-            .to_dummies()
+            .to_dummies(sep.as_deref())
             .map_err(RbPolarsErr::from)?;
         Ok(df.into())
     }
@@ -1095,8 +1097,8 @@ pub fn to_series_collection(rs: RArray) -> RbResult<Vec<Series>> {
     Ok(series)
 }
 
-pub fn to_rbseries_collection(s: Vec<Series>) -> Vec<RbSeries> {
-    s.into_iter().map(RbSeries::new).collect()
+pub fn to_rbseries_collection(s: Vec<Series>) -> RArray {
+    RArray::from_iter(s.into_iter().map(RbSeries::new))
 }
 
 impl RbSeries {

data/lib/polars/cat_expr.rb

@@ -36,13 +36,9 @@ module Polars
     #   # │ cat  ┆ i64  │
     #   # ╞══════╪══════╡
     #   # │ a    ┆ 2    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ b    ┆ 3    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ k    ┆ 2    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ z    ┆ 1    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ z    ┆ 3    │
     #   # └──────┴──────┘
     def set_ordering(ordering)

data/lib/polars/cat_name_space.rb

@@ -38,13 +38,9 @@ module Polars
     #   # │ cat  ┆ i64  │
     #   # ╞══════╪══════╡
     #   # │ a    ┆ 2    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ b    ┆ 3    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ k    ┆ 2    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ z    ┆ 1    │
-    #   # ├╌╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ z    ┆ 3    │
     #   # └──────┴──────┘
     def set_ordering(ordering)

data/lib/polars/convert.rb

@@ -24,7 +24,6 @@ module Polars
     #   # │ i64 ┆ i64 │
     #   # ╞═════╪═════╡
     #   # │ 1   ┆ 3   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 2   ┆ 4   │
     #   # └─────┴─────┘
     def from_hash(data, columns: nil)
@@ -54,9 +53,7 @@ module Polars
     #   # │ i64 ┆ i64 │
     #   # ╞═════╪═════╡
     #   # │ 1   ┆ 4   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 2   ┆ 5   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 3   ┆ 6   │
     #   # └─────┴─────┘
     #
@@ -70,9 +67,7 @@ module Polars
     #   # │ i32 ┆ i64 │
     #   # ╞═════╪═════╡
     #   # │ 1   ┆ 4   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 2   ┆ 5   │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 3   ┆ 6   │
     #   # └─────┴─────┘
     #
@@ -85,9 +80,7 @@ module Polars
     #   # │ i64 ┆ i64 ┆ i32  │
     #   # ╞═════╪═════╪══════╡
     #   # │ 1   ┆ 4   ┆ null │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ 2   ┆ 5   ┆ null │
-    #   # ├╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┤
     #   # │ 3   ┆ 6   ┆ null │
     #   # └─────┴─────┴──────┘
     # def from_hashes(hashes, infer_schema_length: 50, schema: nil)