Adding baseline policy evaluator

src/policies/PolicyEvaluator.py

@@ -35,10 +35,14 @@ class PolicyEvaluator:
         ]
         return imbalance_prices_day["Positive imbalance price"].values
 
-    def evaluate_for_date(self, date, idx_samples, test_loader):
-        charge_thresholds = np.arange(-100, 250, 25)
-        discharge_thresholds = np.arange(-100, 250, 25)
-
+    def evaluate_for_date(
+        self,
+        date,
+        idx_samples,
+        test_loader,
+        charge_thresholds=np.arange(-100, 250, 25),
+        discharge_thresholds=np.arange(-100, 250, 25),
+    ):
         idx = test_loader.dataset.get_idx_for_date(date.date())
 
         print("Evaluated for idx: ", idx)
@@ -95,7 +99,6 @@ class PolicyEvaluator:
     def evaluate_test_set(self, idx_samples, test_loader):
         self.profits = []
         try:
-            print(self.dates)
             for date in tqdm(self.dates):
                 self.evaluate_for_date(date, idx_samples, test_loader)
         except KeyboardInterrupt: