simplify

pola-rs · Jun 5, 2024 · f6d31f8 · f6d31f8
1 parent 42cdc0f
commit f6d31f8
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/crates/polars-plan/src/dsl/function_expr/temporal.rs b/crates/polars-plan/src/dsl/function_expr/temporal.rs
@@ -197,16 +197,15 @@ fn apply_offsets_to_datetime(
                 let offset = &Duration::parse(offset);
                 if offset.is_constant_duration(datetime.time_zone().as_deref()) {
                     // fastpath!
-                    let duration = match datetime.time_unit() {
+                    let mut duration = match datetime.time_unit() {
                         TimeUnit::Milliseconds => offset.duration_ms(),
                         TimeUnit::Microseconds => offset.duration_us(),
                         TimeUnit::Nanoseconds => offset.duration_ns(),
                     };
                     if offset.negative() {
-                        Ok(datetime.0.apply_values(|v| v - duration))
-                    } else {
-                        Ok(datetime.0.apply_values(|v| v + duration))
+                        duration = -duration;
                     }
+                    Ok(datetime.0.apply_values(|v| v + duration))
                 } else {
                     let offset_fn = match datetime.time_unit() {
                         TimeUnit::Milliseconds => Duration::add_ms,