@@ -1152,7 +1152,7 @@ def sell_prices(self):
1152
1152
return self ._sell_prices
1153
1153
1154
1154
@sell_prices .setter
1155
- def sell_prices (self , value : np .ndarray [Any , float ]):
1155
+ def sell_prices (self , value : np .ndarray [Any , np . dtype [ np . float64 ] ]):
1156
1156
if isinstance (value , np .ndarray ) and is_real_numbers (value ):
1157
1157
if value .shape != (YEAR_HOURS ,) and value .shape != (self ._project_hour_num ,):
1158
1158
raise ValueError (f"Prices shape should be ({ YEAR_HOURS } ,) or ({ self ._project_hour_num } , ), prices"
@@ -1172,7 +1172,7 @@ def buy_prices(self):
1172
1172
return self ._buy_prices
1173
1173
1174
1174
@buy_prices .setter
1175
- def buy_prices (self , value : np .ndarray [Any , float ]):
1175
+ def buy_prices (self , value : np .ndarray [Any , np . dtype [ np . float64 ] ]):
1176
1176
if isinstance (value , np .ndarray ) and is_real_numbers (value ):
1177
1177
if value .shape != (YEAR_HOURS ,) and value .shape != (self ._project_hour_num , ):
1178
1178
raise ValueError (f"Prices shape should be ({ YEAR_HOURS } ,) or ({ self ._project_hour_num } , ), prices"
@@ -1200,7 +1200,7 @@ def tariff_table(self, value: np.ndarray[Any, np.dtype[np.float64]]):
1200
1200
raise ValueError ("Tariff table should be of shape (7, 12, 24)" )
1201
1201
self ._tariff_table = value
1202
1202
1203
- def _get_action_from_obs (self , obs : np .ndarray [Any , float ]):
1203
+ def _get_action_from_obs (self , obs : np .ndarray [Any , np . dtype [ np . float64 ] ]):
1204
1204
"""
1205
1205
get action(s) from model given an observation
1206
1206
0 commit comments