Fix PSO constructor

Author: seofernando25

scripts/run_experiment.py

@@ -9,11 +9,12 @@
 from pathlib import Path
 from typing import Any, Dict, Literal
 
+# Add project root to path for imports
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
 from models.decision_tree import DecisionTreeModel
 from models.knn import KNNModel
 
-# Add project root to path for imports
-sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
 
 import numpy as np
 import torch

search/particle_swarm.py

@@ -189,7 +189,12 @@ def __init__(
         c2: float = 1.5,
     ) -> None:
         super().__init__(param_space, evaluate_fn, metric_key, seed)
-        # Vector to param space transformer
+        self.n_jobs = n_jobs if n_jobs is not None else -1
+        self.n_particles = n_particles
+        self.w = w
+        self.c1 = c1
+        self.c2 = c2
+        self._rng = random.Random(seed)
         self.transformer = ParameterTransformer(self.param_space)
 
     def run(