1
1
use arity:: try_binary_elementwise;
2
2
use chrono:: { LocalResult , NaiveDateTime , TimeZone } ;
3
- use pyo3_polars:: export:: polars_core:: datatypes:: time_zone:: parse_time_zone;
4
3
use polars:: prelude:: * ;
5
4
use polars_arrow:: legacy:: time_zone:: Tz ;
5
+ use pyo3_polars:: export:: polars_core:: datatypes:: time_zone:: parse_time_zone;
6
+ use pyo3_polars:: export:: polars_core:: datatypes:: TimeZone as PolarsTimeZone ;
6
7
use pyo3_polars:: export:: polars_core:: utils:: arrow:: legacy:: kernels:: Ambiguous ;
7
8
use pyo3_polars:: export:: polars_core:: utils:: arrow:: temporal_conversions:: {
8
9
timestamp_ms_to_datetime, timestamp_ns_to_datetime, timestamp_us_to_datetime,
9
10
} ;
10
11
use std:: str:: FromStr ;
11
- use pyo3_polars:: export:: polars_core:: datatypes:: TimeZone as PolarsTimeZone ;
12
12
13
13
fn naive_utc_to_naive_local_in_new_time_zone (
14
14
from_tz : & Tz ,
@@ -126,9 +126,7 @@ pub fn elementwise_from_local_datetime(
126
126
) )
127
127
} )
128
128
}
129
- _ => {
130
- Ok ( datetime. apply ( |_| None ) )
131
- }
129
+ _ => Ok ( datetime. apply ( |_| None ) ) ,
132
130
} ,
133
131
_ => try_binary_elementwise ( datetime, from_tz, |timestamp_opt, from_tz_opt| {
134
132
match ( timestamp_opt, from_tz_opt) {
@@ -143,6 +141,9 @@ pub fn elementwise_from_local_datetime(
143
141
}
144
142
} ) ,
145
143
} ;
146
- let out = out?. into_datetime ( datetime. time_unit ( ) , PolarsTimeZone :: opt_try_new ( Some ( out_tz) ) ?) ;
144
+ let out = out?. into_datetime (
145
+ datetime. time_unit ( ) ,
146
+ PolarsTimeZone :: opt_try_new ( Some ( out_tz) ) ?,
147
+ ) ;
147
148
Ok ( out)
148
149
}
0 commit comments