File tree Expand file tree Collapse file tree 1 file changed +8
-9
lines changed
Expand file tree Collapse file tree 1 file changed +8
-9
lines changed Original file line number Diff line number Diff line change @@ -377,15 +377,14 @@ def create_input(
377377 (ndates == nlons == nlats == nalts), then the shape is (ndates,).
378378 """
379379 # Turn everything into arrays
380- dates_arr : npt .NDArray [np .datetime64 ] = np .atleast_1d (
381- np .array (dates , dtype = np .datetime64 )
382- )
383- dyear : npt .NDArray [np .datetime64 ] = (
384- dates_arr .astype ("datetime64[D]" ) - dates_arr .astype ("datetime64[Y]" )
385- ).astype (float ) + 1 # DOY 1-366
386- dseconds : npt .NDArray [np .datetime64 ] = (
387- dates_arr .astype ("datetime64[s]" ) - dates_arr .astype ("datetime64[D]" )
388- ).astype (float )
380+ dates_arr : npt .NDArray [np .datetime64 ] = np .atleast_1d (dates ).astype (np .datetime64 )
381+ dates_arr_y = dates_arr .astype ("datetime64[Y]" )
382+ dates_arr_d = dates_arr .astype ("datetime64[D]" )
383+ dates_arr_s = dates_arr .astype ("datetime64[s]" )
384+
385+ # dyear is DOY 1-366
386+ dyear : npt .NDArray [float ] = (dates_arr_d - dates_arr_y ).astype (float ) + 1
387+ dseconds : npt .NDArray [float ] = (dates_arr_s - dates_arr_d ).astype (float )
389388 # TODO: Make it a continuous day of year?
390389 # The new code mentions it should be and accepts float, but the
391390 # regression tests indicate it should still be integer DOY
You can’t perform that action at this time.
0 commit comments