polars-df 0.9.0 → 0.11.0

data/lib/polars/date_time_expr.rb

@@ -215,6 +215,7 @@ module Polars
         offset = "0ns"
       end
 
+      every = Utils.parse_as_expression(every, str_as_lit: true)
       Utils.wrap_expr(
         _rbexpr.dt_round(
           Utils._timedelta_to_pl_duration(every),
@@ -1027,14 +1028,20 @@ module Polars
     # Different from `convert_time_zone`, this will also modify
     # the underlying timestamp,
     #
-    # @param tz [String]
-    #   Time zone for the `Datetime` Series.
+    # @param time_zone [String]
+    #     Time zone for the `Datetime` Series. Pass `nil` to unset time zone.
+    # @param use_earliest [Boolean]
+    #     Determine how to deal with ambiguous datetimes.
+    # @param ambiguous [String]
+    #     Determine how to deal with ambiguous datetimes.
+    # @param non_existent [String]
+    #     Determine how to deal with non-existent datetimes.
     #
     # @return [Expr]
-    def replace_time_zone(tz, use_earliest: nil, ambiguous: "raise")
+    def replace_time_zone(time_zone, use_earliest: nil, ambiguous: "raise", non_existent: "raise")
       ambiguous = Utils.rename_use_earliest_to_ambiguous(use_earliest, ambiguous)
       ambiguous = Polars.lit(ambiguous) unless ambiguous.is_a?(Expr)
-      Utils.wrap_expr(_rbexpr.dt_replace_time_zone(tz, ambiguous._rbexpr))
+      Utils.wrap_expr(_rbexpr.dt_replace_time_zone(time_zone, ambiguous._rbexpr, non_existent))
     end
 
     # Extract the days from a Duration type.

data/lib/polars/date_time_name_space.rb

@@ -66,6 +66,8 @@ module Polars
       if !out.nil?
         if s.dtype == Date
           return Utils._to_ruby_date(out.to_i)
+        elsif [Datetime, Duration, Time].include?(s.dtype)
+          return out
         else
           return Utils._to_ruby_datetime(out.to_i, s.time_unit)
         end
@@ -93,10 +95,12 @@ module Polars
     #   # => 2001-01-02 00:00:00 UTC
     def mean
       s = Utils.wrap_s(_s)
-      out = s.mean.to_i
+      out = s.mean
       if !out.nil?
         if s.dtype == Date
           return Utils._to_ruby_date(out.to_i)
+        elsif [Datetime, Duration, Time].include?(s.dtype)
+          return out
         else
           return Utils._to_ruby_datetime(out.to_i, s.time_unit)
         end
@@ -910,8 +914,14 @@ module Polars
     # Different from `with_time_zone`, this will also modify
     # the underlying timestamp.
     #
-    # @param tz [String]
-    #   Time zone for the `Datetime` Series.
+    # @param time_zone [String]
+    #     Time zone for the `Datetime` Series. Pass `nil` to unset time zone.
+    # @param use_earliest [Boolean]
+    #     Determine how to deal with ambiguous datetimes.
+    # @param ambiguous [String]
+    #     Determine how to deal with ambiguous datetimes.
+    # @param non_existent [String]
+    #     Determine how to deal with non-existent datetimes.
     #
     # @return [Series]
     #
@@ -982,7 +992,7 @@ module Polars
     #   #         1585717200
     #   #         1588309200
     #   # ]
-    def replace_time_zone(tz)
+    def replace_time_zone(time_zone, use_earliest: nil, ambiguous: "raise", non_existent: "raise")
       super
     end
 

data/lib/polars/dynamic_group_by.rb

@@ -32,7 +32,7 @@ module Polars
       @start_by = start_by
     end
 
-    def agg(aggs)
+    def agg(*aggs, **named_aggs)
       @df.lazy
         .group_by_dynamic(
           @time_column,
@@ -45,7 +45,7 @@ module Polars
           by: @by,
           start_by: @start_by
         )
-        .agg(aggs)
+        .agg(*aggs, **named_aggs)
         .collect(no_optimization: true, string_cache: false)
     end
   end

data/lib/polars/exceptions.rb

@@ -3,6 +3,10 @@ module Polars
   # Base class for all Polars errors.
   class Error < StandardError; end
 
+  # @private
+  # Exception raised when an operation is not allowed (or possible) against a given object or data structure.
+  class InvalidOperationError < Error; end
+
   # @private
   # Exception raised when an unsupported testing assert is made.
   class InvalidAssert < Error; end