polars-df 0.1.5 → 0.2.0

data/ext/polars/src/lib.rs

@@ -56,6 +56,7 @@ fn series() -> RClass {
 #[magnus::init]
 fn init() -> RbResult<()> {
     let module = module();
+    module.define_singleton_method("_dtype_cols", function!(dtype_cols, 1))?;
     module.define_singleton_method("_rb_duration", function!(rb_duration, 8))?;
     module.define_singleton_method("_concat_df", function!(concat_df, 1))?;
     module.define_singleton_method("_concat_lf", function!(concat_lf, 3))?;
@@ -289,9 +290,9 @@ fn init() -> RbResult<()> {
     class.define_method("cumprod", method!(RbExpr::cumprod, 1))?;
     class.define_method("product", method!(RbExpr::product, 0))?;
     class.define_method("shrink_dtype", method!(RbExpr::shrink_dtype, 0))?;
-    class.define_method("str_parse_date", method!(RbExpr::str_parse_date, 3))?;
-    class.define_method("str_parse_datetime", method!(RbExpr::str_parse_datetime, 3))?;
-    class.define_method("str_parse_time", method!(RbExpr::str_parse_time, 3))?;
+    class.define_method("str_parse_date", method!(RbExpr::str_parse_date, 4))?;
+    class.define_method("str_parse_datetime", method!(RbExpr::str_parse_datetime, 5))?;
+    class.define_method("str_parse_time", method!(RbExpr::str_parse_time, 4))?;
     class.define_method("str_strip", method!(RbExpr::str_strip, 1))?;
     class.define_method("str_rstrip", method!(RbExpr::str_rstrip, 1))?;
     class.define_method("str_lstrip", method!(RbExpr::str_lstrip, 1))?;
@@ -382,7 +383,7 @@ fn init() -> RbResult<()> {
     class.define_method("suffix", method!(RbExpr::suffix, 1))?;
     class.define_method("map_alias", method!(RbExpr::map_alias, 1))?;
     class.define_method("exclude", method!(RbExpr::exclude, 1))?;
-    class.define_method("interpolate", method!(RbExpr::interpolate, 0))?;
+    class.define_method("interpolate", method!(RbExpr::interpolate, 1))?;
     class.define_method("rolling_sum", method!(RbExpr::rolling_sum, 6))?;
     class.define_method("rolling_min", method!(RbExpr::rolling_min, 6))?;
     class.define_method("rolling_max", method!(RbExpr::rolling_max, 6))?;
@@ -410,7 +411,7 @@ fn init() -> RbResult<()> {
     class.define_method("lst_slice", method!(RbExpr::lst_slice, 2))?;
     class.define_method("lst_eval", method!(RbExpr::lst_eval, 2))?;
     class.define_method("cumulative_eval", method!(RbExpr::cumulative_eval, 3))?;
-    class.define_method("lst_to_struct", method!(RbExpr::lst_to_struct, 2))?;
+    class.define_method("lst_to_struct", method!(RbExpr::lst_to_struct, 3))?;
     class.define_method("rank", method!(RbExpr::rank, 2))?;
     class.define_method("diff", method!(RbExpr::diff, 2))?;
     class.define_method("pct_change", method!(RbExpr::pct_change, 1))?;
@@ -507,7 +508,7 @@ fn init() -> RbResult<()> {
     class.define_method("select", method!(RbLazyFrame::select, 1))?;
     class.define_method("groupby", method!(RbLazyFrame::groupby, 2))?;
     class.define_method("groupby_rolling", method!(RbLazyFrame::groupby_rolling, 5))?;
-    class.define_method("groupby_dynamic", method!(RbLazyFrame::groupby_dynamic, 8))?;
+    class.define_method("groupby_dynamic", method!(RbLazyFrame::groupby_dynamic, 9))?;
     class.define_method("with_context", method!(RbLazyFrame::with_context, 1))?;
     class.define_method("join_asof", method!(RbLazyFrame::join_asof, 11))?;
     class.define_method("join", method!(RbLazyFrame::join, 7))?;
@@ -792,6 +793,15 @@ fn init() -> RbResult<()> {
     Ok(())
 }
 
+fn dtype_cols(dtypes: RArray) -> RbResult<RbExpr> {
+    let dtypes = dtypes
+        .each()
+        .map(|v| v?.try_convert::<Wrap<DataType>>())
+        .collect::<RbResult<Vec<Wrap<DataType>>>>()?;
+    let dtypes = vec_extract_wrapped(dtypes);
+    Ok(crate::lazy::dsl::dtype_cols(dtypes))
+}
+
 #[allow(clippy::too_many_arguments)]
 fn rb_duration(
     days: Option<&RbExpr>,
@@ -927,7 +937,7 @@ fn collect_all(lfs: RArray) -> RbResult<Vec<RbDataFrame>> {
         .map(|v| v?.try_convert::<&RbLazyFrame>())
         .collect::<RbResult<Vec<&RbLazyFrame>>>()?;
 
-    let out = polars_core::POOL.install(|| {
+    polars_core::POOL.install(|| {
         lfs.par_iter()
             .map(|lf| {
                 let df = lf.ldf.clone().collect()?;
@@ -935,9 +945,7 @@ fn collect_all(lfs: RArray) -> RbResult<Vec<RbDataFrame>> {
             })
             .collect::<polars_core::error::PolarsResult<Vec<_>>>()
             .map_err(RbPolarsErr::from)
-    });
-
-    Ok(out?)
+    })
 }
 
 fn rb_date_range(
@@ -956,7 +964,7 @@ fn rb_date_range(
         Duration::parse(&every),
         closed.0,
         tu.0,
-        tz,
+        tz.as_ref(),
     )
     .into_series()
     .into()

data/ext/polars/src/series.rs

@@ -144,7 +144,7 @@ impl RbSeries {
     }
 
     pub fn get_fmt(&self, index: usize, str_lengths: usize) -> String {
-        let val = format!("{}", self.series.borrow().get(index));
+        let val = format!("{}", self.series.borrow().get(index).unwrap());
         if let DataType::Utf8 | DataType::Categorical(_) = self.series.borrow().dtype() {
             let v_trunc = &val[..val
                 .char_indices()
@@ -172,8 +172,8 @@ impl RbSeries {
         }
     }
 
-    pub fn get_idx(&self, idx: usize) -> Value {
-        Wrap(self.series.borrow().get(idx)).into()
+    pub fn get_idx(&self, idx: usize) -> RbResult<Value> {
+        Ok(Wrap(self.series.borrow().get(idx).map_err(RbPolarsErr::from)?).into())
     }
 
     pub fn bitand(&self, other: &RbSeries) -> RbResult<Self> {
@@ -247,16 +247,37 @@ impl RbSeries {
         }
     }
 
-    pub fn max(&self) -> Value {
-        Wrap(self.series.borrow().max_as_series().get(0)).into()
+    pub fn max(&self) -> RbResult<Value> {
+        Ok(Wrap(
+            self.series
+                .borrow()
+                .max_as_series()
+                .get(0)
+                .map_err(RbPolarsErr::from)?,
+        )
+        .into())
     }
 
-    pub fn min(&self) -> Value {
-        Wrap(self.series.borrow().min_as_series().get(0)).into()
+    pub fn min(&self) -> RbResult<Value> {
+        Ok(Wrap(
+            self.series
+                .borrow()
+                .min_as_series()
+                .get(0)
+                .map_err(RbPolarsErr::from)?,
+        )
+        .into())
     }
 
-    pub fn sum(&self) -> Value {
-        Wrap(self.series.borrow().sum_as_series().get(0)).into()
+    pub fn sum(&self) -> RbResult<Value> {
+        Ok(Wrap(
+            self.series
+                .borrow()
+                .sum_as_series()
+                .get(0)
+                .map_err(RbPolarsErr::from)?,
+        )
+        .into())
     }
 
     pub fn n_chunks(&self) -> usize {
@@ -522,7 +543,8 @@ impl RbSeries {
                 .borrow()
                 .quantile_as_series(quantile, interpolation.0)
                 .map_err(|_| RbValueError::new_err("invalid quantile".into()))?
-                .get(0),
+                .get(0)
+                .unwrap_or(AnyValue::Null),
         )
         .into())
     }

data/lib/polars/data_frame.rb

@@ -354,7 +354,7 @@ module Polars
     #     }
     #   )
     #   df.dtypes
-    #   # => [:i64, :f64, :str]
+    #   # => [Polars::Int64, Polars::Float64, Polars::Utf8]
     def dtypes
       _df.dtypes
     end
@@ -372,7 +372,7 @@ module Polars
     #     }
     #   )
     #   df.schema
-    #   # => {"foo"=>:i64, "bar"=>:f64, "ham"=>:str}
+    #   # => {"foo"=>Polars::Int64, "bar"=>Polars::Float64, "ham"=>Polars::Utf8}
     def schema
       columns.zip(dtypes).to_h
     end
@@ -1178,25 +1178,25 @@ module Polars
     #   df.describe
     #   # =>
     #   # shape: (7, 6)
-    #   # ┌────────────┬──────────┬──────────┬──────┬──────┬──────┐
-    #   # │ describe   ┆ a        ┆ b        ┆ c    ┆ d    ┆ e    │
-    #   # │ ---        ┆ ---      ┆ ---      ┆ ---  ┆ ---  ┆ ---  │
-    #   # │ str        ┆ f64      ┆ f64      ┆ f64  ┆ str  ┆ str  │
-    #   # ╞════════════╪══════════╪══════════╪══════╪══════╪══════╡
-    #   # │ count      ┆ 3.0      ┆ 3.0      ┆ 3.0  ┆ 3    ┆ 3    │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ null_count ┆ 0.0      ┆ 1.0      ┆ 0.0  ┆ 1    ┆ 1    │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ mean       ┆ 2.266667 ┆ 4.5      ┆ null ┆ null ┆ null │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ std        ┆ 1.101514 ┆ 0.707107 ┆ null ┆ null ┆ null │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ min        ┆ 1.0      ┆ 4.0      ┆ 0.0  ┆ b    ┆ eur  │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ max        ┆ 3.0      ┆ 5.0      ┆ 1.0  ┆ c    ┆ usd  │
-    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ median     ┆ 2.8      ┆ 4.5      ┆ null ┆ null ┆ null │
-    #   # └────────────┴──────────┴──────────┴──────┴──────┴──────┘
+    #   # ┌────────────┬──────────┬──────────┬──────────┬──────┬──────┐
+    #   # │ describe   ┆ a        ┆ b        ┆ c        ┆ d    ┆ e    │
+    #   # │ ---        ┆ ---      ┆ ---      ┆ ---      ┆ ---  ┆ ---  │
+    #   # │ str        ┆ f64      ┆ f64      ┆ f64      ┆ str  ┆ str  │
+    #   # ╞════════════╪══════════╪══════════╪══════════╪══════╪══════╡
+    #   # │ count      ┆ 3.0      ┆ 3.0      ┆ 3.0      ┆ 3    ┆ 3    │
+    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ null_count ┆ 0.0      ┆ 1.0      ┆ 0.0      ┆ 1    ┆ 1    │
+    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ mean       ┆ 2.266667 ┆ 4.5      ┆ 0.666667 ┆ null ┆ null │
+    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ std        ┆ 1.101514 ┆ 0.707107 ┆ 0.57735  ┆ null ┆ null │
+    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ min        ┆ 1.0      ┆ 4.0      ┆ 0.0      ┆ b    ┆ eur  │
+    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ max        ┆ 3.0      ┆ 5.0      ┆ 1.0      ┆ c    ┆ usd  │
+    #   # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ median     ┆ 2.8      ┆ 4.5      ┆ 1.0      ┆ null ┆ null │
+    #   # └────────────┴──────────┴──────────┴──────────┴──────┴──────┘
     def describe
       describe_cast = lambda do |stat|
         columns = []
@@ -2074,7 +2074,8 @@ module Polars
       truncate: true,
       include_boundaries: false,
       closed: "left",
-      by: nil
+      by: nil,
+      start_by: "window"
     )
       DynamicGroupBy.new(
         self,
@@ -2085,7 +2086,8 @@ module Polars
         truncate,
         include_boundaries,
         closed,
-        by
+        by,
+        start_by
       )
     end
 

data/lib/polars/data_types.rb

@@ -0,0 +1,122 @@
+module Polars
+  # Base class for all Polars data types.
+  class DataType
+  end
+
+  # 8-bit signed integer type.
+  class Int8 < DataType
+  end
+
+  # 16-bit signed integer type.
+  class Int16 < DataType
+  end
+
+  # 32-bit signed integer type.
+  class Int32 < DataType
+  end
+
+  # 64-bit signed integer type.
+  class Int64 < DataType
+  end
+
+  # 8-bit unsigned integer type.
+  class UInt8 < DataType
+  end
+
+  # 16-bit unsigned integer type.
+  class UInt16 < DataType
+  end
+
+  # 32-bit unsigned integer type.
+  class UInt32 < DataType
+  end
+
+  # 64-bit unsigned integer type.
+  class UInt64 < DataType
+  end
+
+  # 32-bit floating point type.
+  class Float32 < DataType
+  end
+
+  # 64-bit floating point type.
+  class Float64 < DataType
+  end
+
+  # Boolean type.
+  class Boolean < DataType
+  end
+
+  # UTF-8 encoded string type.
+  class Utf8 < DataType
+  end
+
+  # Binary type.
+  class Binary < DataType
+  end
+
+  # Type representing Null / None values.
+  class Null < DataType
+  end
+
+  # Type representing Datatype values that could not be determined statically.
+  class Unknown < DataType
+  end
+
+  # Nested list/array type.
+  class List < DataType
+    def initialize(inner)
+      @inner = Utils.rb_type_to_dtype(inner)
+    end
+  end
+
+  # Calendar date type.
+  class Date < DataType
+  end
+
+  # Calendar date and time type.
+  class Datetime < DataType
+    def initialize(time_unit = "us", time_zone = nil)
+      @tu = time_unit || "us"
+      @time_zone = time_zone
+    end
+  end
+
+  # Time duration/delta type.
+  class Duration < DataType
+    def initialize(time_unit = "us")
+      @tu = time_unit
+    end
+  end
+
+  # Time of day type.
+  class Time < DataType
+  end
+
+  # Type for wrapping arbitrary Python objects.
+  class Object < DataType
+  end
+
+  # A categorical encoding of a set of strings.
+  class Categorical < DataType
+  end
+
+  # Definition of a single field within a `Struct` DataType.
+  class Field < DataType
+    def initialize(name, dtype)
+      @name = name
+      @dtype = Utils.rb_type_to_dtype(dtype)
+    end
+  end
+
+  # Struct composite type.
+  class Struct < DataType
+    def initialize(fields)
+      if fields.is_a?(Hash)
+        @fields = fields.map { |n, d| Field.new(n, d) }
+      else
+        @fields = fields
+      end
+    end
+  end
+end

data/lib/polars/date_time_expr.rb

@@ -465,7 +465,7 @@ module Polars
     #
     # Applies to Date and Datetime columns.
     #
-    # Returns the weekday number where monday = 0 and sunday = 6
+    # Returns the ISO weekday number where monday = 1 and sunday = 7
     #
     # @return [Expr]
     #
@@ -502,11 +502,11 @@ module Polars
     #   # │ ---     ┆ ---          ┆ ---         │
     #   # │ u32     ┆ u32          ┆ u32         │
     #   # ╞═════════╪══════════════╪═════════════╡
-    #   # │ 0       ┆ 1            ┆ 1           │
+    #   # │ 1       ┆ 1            ┆ 1           │
     #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-    #   # │ 3       ┆ 4            ┆ 4           │
+    #   # │ 4       ┆ 4            ┆ 4           │
     #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-    #   # │ 6       ┆ 7            ┆ 7           │
+    #   # │ 7       ┆ 7            ┆ 7           │
     #   # └─────────┴──────────────┴─────────────┘
     def weekday
       Utils.wrap_expr(_rbexpr.weekday)
@@ -554,11 +554,11 @@ module Polars
     #   # │ ---     ┆ ---          ┆ ---         │
     #   # │ u32     ┆ u32          ┆ u32         │
     #   # ╞═════════╪══════════════╪═════════════╡
-    #   # │ 0       ┆ 1            ┆ 1           │
+    #   # │ 1       ┆ 1            ┆ 1           │
     #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-    #   # │ 3       ┆ 4            ┆ 4           │
+    #   # │ 4       ┆ 4            ┆ 4           │
     #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-    #   # │ 6       ┆ 7            ┆ 7           │
+    #   # │ 7       ┆ 7            ┆ 7           │
     #   # └─────────┴──────────────┴─────────────┘
     def day
       Utils.wrap_expr(_rbexpr.day)
@@ -606,11 +606,11 @@ module Polars
     #   # │ ---     ┆ ---          ┆ ---         │
     #   # │ u32     ┆ u32          ┆ u32         │
     #   # ╞═════════╪══════════════╪═════════════╡
-    #   # │ 0       ┆ 1            ┆ 1           │
+    #   # │ 1       ┆ 1            ┆ 1           │
     #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-    #   # │ 3       ┆ 4            ┆ 4           │
+    #   # │ 4       ┆ 4            ┆ 4           │
     #   # ├╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
-    #   # │ 6       ┆ 7            ┆ 7           │
+    #   # │ 7       ┆ 7            ┆ 7           │
     #   # └─────────┴──────────────┴─────────────┘
     def ordinal_day
       Utils.wrap_expr(_rbexpr.ordinal_day)

data/lib/polars/date_time_name_space.rb

@@ -317,7 +317,7 @@ module Polars
     #
     # Applies to Date and Datetime columns.
     #
-    # Returns the weekday number where monday = 0 and sunday = 6
+    # Returns the ISO weekday number where monday = 1 and sunday = 7
     #
     # @return [Series]
     #
@@ -344,13 +344,13 @@ module Polars
     #   # shape: (7,)
     #   # Series: '' [u32]
     #   # [
-    #   #         0
     #   #         1
     #   #         2
     #   #         3
     #   #         4
     #   #         5
     #   #         6
+    #   #         7
     #   # ]
     def weekday
       super
@@ -973,9 +973,9 @@ module Polars
     #   # shape: (3,)
     #   # Series: 'NYC' [datetime[μs, America/New_York]]
     #   # [
-    #   #         2020-02-29 14:00:00 EST
-    #   #         2020-03-31 15:00:00 EDT
-    #   #         2020-04-30 15:00:00 EDT
+    #   #         2020-03-01 00:00:00 EST
+    #   #         2020-04-01 01:00:00 EDT
+    #   #         2020-05-01 01:00:00 EDT
     #   # ]
     #
     # @example Timestamps have changed after cast_time_zone
@@ -984,9 +984,9 @@ module Polars
     #   # shape: (3,)
     #   # Series: 'NYC' [i64]
     #   # [
-    #   #         1583002800
-    #   #         1585681200
-    #   #         1588273200
+    #   #         1583038800
+    #   #         1585717200
+    #   #         1588309200
     #   # ]
     def cast_time_zone(tz)
       super

data/lib/polars/dynamic_group_by.rb

@@ -13,7 +13,8 @@ module Polars
       truncate,
       include_boundaries,
       closed,
-      by
+      by,
+      start_by
     )
       period = Utils._timedelta_to_pl_duration(period)
       offset = Utils._timedelta_to_pl_duration(offset)
@@ -28,6 +29,7 @@ module Polars
       @include_boundaries = include_boundaries
       @closed = closed
       @by = by
+      @start_by = start_by
     end
 
     def agg(aggs)
@@ -40,7 +42,8 @@ module Polars
           truncate: @truncate,
           include_boundaries: @include_boundaries,
           closed: @closed,
-          by: @by
+          by: @by,
+          start_by: @start_by
         )
         .agg(aggs)
         .collect(no_optimization: true, string_cache: false)

data/lib/polars/expr.rb

@@ -2486,7 +2486,8 @@ module Polars
     #   # │ 1.5 │
     #   # └─────┘
     def quantile(quantile, interpolation: "nearest")
-      wrap_expr(_rbexpr.quantile(quantile, interpolation))
+      quantile = Utils.expr_to_lit_or_expr(quantile, str_to_lit: false)
+      wrap_expr(_rbexpr.quantile(quantile._rbexpr, interpolation))
     end
 
     # Filter a single column.
@@ -3151,8 +3152,8 @@ module Polars
     #   # ├╌╌╌╌╌┼╌╌╌╌╌┤
     #   # │ 3   ┆ 3.0 │
     #   # └─────┴─────┘
-    def interpolate
-      wrap_expr(_rbexpr.interpolate)
+    def interpolate(method: "linear")
+      wrap_expr(_rbexpr.interpolate(method))
     end
 
     # Apply a rolling min (moving min) over the values in this array.

data/lib/polars/functions.rb

@@ -438,8 +438,8 @@ module Polars
 
     def _ensure_datetime(value)
       is_date_type = false
-      if !value.is_a?(DateTime)
-        value = DateTime.new(value.year, value.month, value.day)
+      if !value.is_a?(::DateTime)
+        value = ::DateTime.new(value.year, value.month, value.day)
         is_date_type = true
       end
       [value, is_date_type]

data/lib/polars/group_by.rb

@@ -334,17 +334,17 @@ module Polars
     #   df.groupby("d", maintain_order: true).min
     #   # =>
     #   # shape: (3, 4)
-    #   # ┌────────┬─────┬──────┬─────┐
-    #   # │ d      ┆ a   ┆ b    ┆ c   │
-    #   # │ ---    ┆ --- ┆ ---  ┆ --- │
-    #   # │ str    ┆ i64 ┆ f64  ┆ u32 │
-    #   # ╞════════╪═════╪══════╪═════╡
-    #   # │ Apple  ┆ 1   ┆ 0.5  ┆ 0   │
-    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌┤
-    #   # │ Orange ┆ 2   ┆ 0.5  ┆ 1   │
-    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌┤
-    #   # │ Banana ┆ 4   ┆ 13.0 ┆ 0   │
-    #   # └────────┴─────┴──────┴─────┘
+    #   # ┌────────┬─────┬──────┬───────┐
+    #   # │ d      ┆ a   ┆ b    ┆ c     │
+    #   # │ ---    ┆ --- ┆ ---  ┆ ---   │
+    #   # │ str    ┆ i64 ┆ f64  ┆ bool  │
+    #   # ╞════════╪═════╪══════╪═══════╡
+    #   # │ Apple  ┆ 1   ┆ 0.5  ┆ false │
+    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
+    #   # │ Orange ┆ 2   ┆ 0.5  ┆ true  │
+    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌╌┤
+    #   # │ Banana ┆ 4   ┆ 13.0 ┆ false │
+    #   # └────────┴─────┴──────┴───────┘
     def min
       agg(Polars.all.min)
     end
@@ -365,17 +365,17 @@ module Polars
     #   df.groupby("d", maintain_order: true).max
     #   # =>
     #   # shape: (3, 4)
-    #   # ┌────────┬─────┬──────┬─────┐
-    #   # │ d      ┆ a   ┆ b    ┆ c   │
-    #   # │ ---    ┆ --- ┆ ---  ┆ --- │
-    #   # │ str    ┆ i64 ┆ f64  ┆ u32 │
-    #   # ╞════════╪═════╪══════╪═════╡
-    #   # │ Apple  ┆ 3   ┆ 10.0 ┆ 1   │
-    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌┤
-    #   # │ Orange ┆ 2   ┆ 0.5  ┆ 1   │
-    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌┤
-    #   # │ Banana ┆ 5   ┆ 14.0 ┆ 1   │
-    #   # └────────┴─────┴──────┴─────┘
+    #   # ┌────────┬─────┬──────┬──────┐
+    #   # │ d      ┆ a   ┆ b    ┆ c    │
+    #   # │ ---    ┆ --- ┆ ---  ┆ ---  │
+    #   # │ str    ┆ i64 ┆ f64  ┆ bool │
+    #   # ╞════════╪═════╪══════╪══════╡
+    #   # │ Apple  ┆ 3   ┆ 10.0 ┆ true │
+    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ Orange ┆ 2   ┆ 0.5  ┆ true │
+    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌┼╌╌╌╌╌╌┤
+    #   # │ Banana ┆ 5   ┆ 14.0 ┆ true │
+    #   # └────────┴─────┴──────┴──────┘
     def max
       agg(Polars.all.max)
     end
@@ -427,17 +427,17 @@ module Polars
     #   df.groupby("d", maintain_order: true).mean
     #   # =>
     #   # shape: (3, 4)
-    #   # ┌────────┬─────┬──────────┬──────┐
-    #   # │ d      ┆ a   ┆ b        ┆ c    │
-    #   # │ ---    ┆ --- ┆ ---      ┆ ---  │
-    #   # │ str    ┆ f64 ┆ f64      ┆ bool │
-    #   # ╞════════╪═════╪══════════╪══════╡
-    #   # │ Apple  ┆ 2.0 ┆ 4.833333 ┆ null │
-    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ Orange ┆ 2.0 ┆ 0.5      ┆ null │
-    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌┤
-    #   # │ Banana ┆ 4.5 ┆ 13.5     ┆ null │
-    #   # └────────┴─────┴──────────┴──────┘
+    #   # ┌────────┬─────┬──────────┬──────────┐
+    #   # │ d      ┆ a   ┆ b        ┆ c        │
+    #   # │ ---    ┆ --- ┆ ---      ┆ ---      │
+    #   # │ str    ┆ f64 ┆ f64      ┆ f64      │
+    #   # ╞════════╪═════╪══════════╪══════════╡
+    #   # │ Apple  ┆ 2.0 ┆ 4.833333 ┆ 0.666667 │
+    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
+    #   # │ Orange ┆ 2.0 ┆ 0.5      ┆ 1.0      │
+    #   # ├╌╌╌╌╌╌╌╌┼╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌┤
+    #   # │ Banana ┆ 4.5 ┆ 13.5     ┆ 0.5      │
+    #   # └────────┴─────┴──────────┴──────────┘
     def mean
       agg(Polars.all.mean)
     end