Bug fix

hawk/analysis/main.py

@@ -85,15 +85,24 @@ def run_baseline_analysis(self):
 
         configs = []
 
-        # Autoregressive baselines
-        for i in range(1, self.tefs_max_lag_target):
+        # Only autoregressive baselines from 1 to the maximum target lag
+        for i in range(1, self.tefs_max_lag_target+1):
             configs.append((f"AR({i})", {self.target_column_name: list(range(1, i + 1))}))
 
-        # With all features
+        # All features without AR
         configs.append(
             (
                 "All features",
-                {feature: self.tefs_features_lags for feature in features_names},
+                {feature: self.tefs_features_lags for feature in features_names if feature != self.target_column_name},
+            )
+        )
+
+        # All features with AR
+        configs.append(
+            (
+                "All features and AR",
+                {feature: self.tefs_features_lags if feature != self.target_column_name
+                    else list(range(1, self.tefs_max_lag_target+1)) for feature in features_names},
             )
         )
 
@@ -169,7 +178,7 @@ def run_tefs_analysis(
     def run_pcmci_analysis(
         self,
     ):
-        lag_options = self.pcmci_features_lags  # max lag
+        lag_options = self.pcmci_features_lags  # list from 0 to max_lag
 
         # Define the tests
         parcorr = ParCorr(significance="analytic")
@@ -180,6 +189,7 @@ def run_pcmci_analysis(
             transform="ranks",
             sig_samples=200,
         )
+        # cmiknn = CMIknn()
 
         # Create the dictionary of tests
         independence_tests = {