Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jul 26, 2024
1 parent 95a719d commit 4124224
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion ocf_datapipes/transform/numpy_batch/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,5 +2,5 @@

from .add_fourier_space_time import AddFourierSpaceTimeIterDataPipe as AddFourierSpaceTime
from .add_topographic_data import AddTopographicDataIterDataPipe as AddTopographicData
from .sun_position import AddSunPositionIterDataPipe as AddSunPosition
from .datetime_features import AddTrigonometricDateTimeIterDataPipe as AddTrigonometricDateTime
from .sun_position import AddSunPositionIterDataPipe as AddSunPosition
8 changes: 4 additions & 4 deletions ocf_datapipes/transform/numpy_batch/datetime_features.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,12 @@


def _get_date_time_in_pi(dt):
day_of_year = (dt - dt.astype('datetime64[Y]')).astype(int)
minute_of_day = (dt - dt.astype('datetime64[D]')).astype(int)
day_of_year = (dt - dt.astype("datetime64[Y]")).astype(int)
minute_of_day = (dt - dt.astype("datetime64[D]")).astype(int)

# converting into positions on sin-cos circle
time_in_pi = (2*np.pi) * (minute_of_day / (24*3600))
date_in_pi = (2*np.pi) * (day_of_year / (365*24*3600))
time_in_pi = (2 * np.pi) * (minute_of_day / (24 * 3600))
date_in_pi = (2 * np.pi) * (day_of_year / (365 * 24 * 3600))

return date_in_pi, time_in_pi

Expand Down

0 comments on commit 4124224

Please sign in to comment.