Made more changes

src/policies/baselines/PerfectBaseline.py

@@ -1,5 +1,5 @@
 from clearml import Task
-from policies.simple_baseline import BaselinePolicy
+from src.policies.simple_baseline import BaselinePolicy
 from src.policies.baselines.YesterdayBaselinePolicyExecutor import (
     YesterdayBaselinePolicyEvaluator,
 )
@@ -14,17 +14,21 @@ class PerfectBaseline(YesterdayBaselinePolicyEvaluator):
     def evaluate_for_date(
         self,
         date,
-        charge_thresholds=np.arange(-100, 250, 25),
-        discharge_thresholds=np.arange(-100, 250, 25),
+        charge_thresholds=np.arange(-300, 300, 5),
+        discharge_thresholds=np.arange(-300, 300, 5),
         penalty: int = 0,
         current_state_of_charge=0.0,
     ):
 
         real_imbalance_prices = self.get_imbanlance_prices_for_date(date.date())
 
+        real_imbalance_prices_tensor = torch.tensor(
+            np.array([real_imbalance_prices]), device="cpu"
+        )
+
         best_charge_thresholds, best_discharge_thresholds = (
             self.baseline_policy.get_optimal_thresholds(
-                real_imbalance_prices,
+                real_imbalance_prices_tensor,
                 charge_thresholds,
                 discharge_thresholds,
                 penalty,
@@ -45,4 +49,6 @@ class PerfectBaseline(YesterdayBaselinePolicyEvaluator):
             best_profit[0][0].item(),
             best_charge_cycles[0][0].item(),
             new_state_of_charge.squeeze(0).item(),
+            best_charge_thresholds.mean(axis=0),
+            best_discharge_thresholds.mean(axis=0),
         )