Fixed policy evaluation for autoregressive

src/policies/baselines/YesterdayBaselinePolicyExecutor.py

@@ -54,14 +54,26 @@ class YesterdayBaselinePolicyEvaluator(PolicyEvaluator):
                 ]
             )
 
-    def evaluate_test_set(self):
+    def evaluate_test_set(self, data_processor):
+
+        if data_processor:
+            filtered_dates = []
+            _, test_loader = data_processor.get_dataloaders()
+            for date in self.dates:
+                try:
+                    test_loader.dataset.get_idx_for_date(date.date())
+                    filtered_dates.append(date)
+                except:
+                    pass
+            self.dates = filtered_dates
+
         self.profits = []
-        try:
-            for date in tqdm(self.dates):
+        for date in tqdm(self.dates):
+            try:
                 self.evaluate_for_date(date)
-        except Exception as e:
-            print(e)
-            pass
+            except Exception as e:
+                print(e)
+                pass
 
         self.profits = pd.DataFrame(
             self.profits,