fixed small mistake in output_calculator.py

Author: pvstorageoptimization

optibess_algorithm/output_calculator.py

@@ -1115,11 +1115,10 @@ def __init__(self, tariff_table: np.ndarray[Any, np.dtype[np.float64]] | None =
         super().__init__(*args, **kwargs)
         self._fill_battery_from_grid = True
         self.tariff_table = tariff_table
+        self._sell_prices = self._buy_prices = None
         if sell_prices is not None:
             self.sell_prices = sell_prices
             self.buy_prices = buy_prices if buy_prices is not None else sell_prices
-        else:
-            self._sell_prices = self._buy_prices = None
         self._model_session = onnxruntime.InferenceSession(os.path.join(os.path.dirname(__file__),
                                                                         "schedule_model.onnx"),
                                                            providers=["CPUExecutionProvider"])
@@ -1246,6 +1245,8 @@ def _get_actions(self, year):
         stride = self._df["pv_output"].values.strides[0]
         split_hourly_power = ast(normalized_hourly_power, (YEAR_DAYS + day_add, DAY_LENGTH),
                                  (DAY_LENGTH * stride, stride))
+        temp = normalized_hourly_power.reshape((YEAR_DAYS + day_add, DAY_LENGTH))
+        print(np.all(np.equal(split_hourly_power, temp)))
         split_hourly_power = np.repeat(split_hourly_power, DAY_LENGTH, axis=0) + pos_encoding
         split_prices = ast(sell_prices, (YEAR_DAYS + day_add, DAY_LENGTH), (DAY_LENGTH, sell_prices.strides[0]))
         max_sell_prices = np.repeat(np.max(np.abs(split_prices), axis=1) + self.EPSILON, DAY_LENGTH)

tests/test_output_calculator.py

@@ -1,5 +1,7 @@
 import unittest
 from unittest.mock import Mock
+
+import onnxruntime
 import pandas as pd
 import numpy as np
 import numpy.testing as nptesting
@@ -789,8 +791,10 @@ def test_run(self):
         output._prod_trans_loss = 0.024
         output._charge_loss = 0.035
         output._grid_bess_loss = 0.04
+        output._model_session = onnxruntime.InferenceSession("output_calculator/test_model.onnx",
+                                                             providers=["CPUExecutionProvider"])
         output.run()
-        expected_result = [-0., -0., -0., -0., -0., -0., -0., 7000., 7000.,
-                           7000., 7000., 7000., 7000., 1978.25, 5345.75, 2786.48, 7000., 6974.29,
-                           6974.29, 3581.76, -0., -0., -0., -0.]
+        expected_result = [-0., -0., -0., -0., -0., -0., -0., 6442.93, 6650.14,
+                           6656.05, 7000., 7000., 6745.98, 6830.97, 5384.28, 4091.53, 7000., 6974.29,
+                           6370.69, -0., -0., -0., -0., -0.]
         nptesting.assert_array_almost_equal(output.output[0][:24], expected_result, 2)