[HOTFIX]: DatetimeWrapper: Added -1 in month cos and sin due to its r…

…ange [1,12] instead of [0,11]
StevenNAT · Apr 17, 2023 · 9b90bee · 9b90bee
1 parent ffc560d
commit 9b90bee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sinergym/utils/wrappers.py b/sinergym/utils/wrappers.py
@@ -405,8 +405,8 @@ def observation(self, obs: np.ndarray) -> np.ndarray:
         new_obs['is_weekend'] = 1.0 if dt.isoweekday() in [6, 7] else 0.0
         new_obs['hour_cos'] = np.cos(2 * np.pi * obs_dict['hour'] / 24)
         new_obs['hour_sin'] = np.sin(2 * np.pi * obs_dict['hour'] / 24)
-        new_obs['month_cos'] = np.cos(2 * np.pi * obs_dict['month'] / 12)
-        new_obs['month_sin'] = np.sin(2 * np.pi * obs_dict['month'] / 12)
+        new_obs['month_cos'] = np.cos(2 * np.pi * (obs_dict['month'] - 1) / 12)
+        new_obs['month_sin'] = np.sin(2 * np.pi * (obs_dict['month'] - 1) / 12)
 
         return np.array(list(new_obs.values()))