update error codes

Author: TheEimer

hydra_plugins/hyper_analysis/ablation_path_sweeper.py

@@ -47,7 +47,7 @@ def __init__(
                 if not np.isnan(target_config[k]):
                     self.target_config[k] = target_config[k]
         else:
-            df = load_data(data_path, performance_key, config_key, variation_key) # noqa: PD901
+            df = load_data(data_path, performance_key, config_key, variation_key)
             self.source_config = df_to_config(configspace, get_overall_best_config(df))
             self.target_config = df_to_config(configspace, get_best_config_per_variation(df, variation))
 

hydra_plugins/hyper_analysis/lpi_sweeper.py

@@ -38,7 +38,7 @@ def __init__(
         if config:
             self.config = dict_to_config(configspace, config)
         else:
-            df = load_data(data_path, performance_key, config_key, variation_key) # noqa: PD901
+            df = load_data(data_path, performance_key, config_key, variation_key)
             self.config = df_to_config(configspace, get_overall_best_config(df))
 
         neighbors = self._get_neighborhood()
@@ -68,7 +68,7 @@ def tell(self, info, value):
 
     def finish_run(self, output_path):
         """Finish the run and calculate LPI scores."""
-        df = pd.read_csv(pathlib.PurePath(output_path, "runhistory.csv")) # noqa: PD901
+        df = pd.read_csv(pathlib.PurePath(output_path, "runhistory.csv"))
         if "seed" not in df.columns:
             df["seed"] = 0
         seeds = df["seed"].unique()

hydra_plugins/hyper_analysis/utils.py

@@ -37,7 +37,7 @@ def get_overall_best_config(df):
 
 def get_best_config_per_variation(df, var, variation_key="env"):
     """Get the best config for a specific variation."""
-    df = df[df[variation_key] == var] # noqa: PD901
+    df = df[df[variation_key] == var]
     return df.loc[df["mean_performance"].idxmax()]
 
 

pyproject.toml

@@ -191,6 +191,7 @@ lint.ignore = [
   "B023",
   "B904",
   "PLC1802",
+  "PD901"
 ]
 
 exclude = [