Address comments

Author: forklady42

src/electrai/configs/MP/config_hpo_tier1.yaml

@@ -46,8 +46,8 @@ data:
   split_file: null
   precision: f32
   batch_size: 1  # fixed
-  # train_workers: 8
-  # val_workers: 2
+  train_workers: 8
+  val_workers: 2
   pin_memory: false
   val_frac: 0.1
   drop_last: false

src/electrai/entrypoints/hpo.py

@@ -1,5 +1,9 @@
 """Hyperparameter optimization entrypoint using Optuna.
 
+Installation: This module requires the optional ``hpo`` extra (optuna and
+optuna-integration). Install with:
+    uv sync --extra hpo
+
 Run with regular python (NOT torchrun):
     uv run python src/electrai/entrypoints/hpo.py --config path/to/config.yaml
 
@@ -159,7 +163,7 @@ def objective(trial: Trial) -> float:
                     "trial_number": trial.number,
                     "num_devices": num_devices,
                 },
-                reinit=True,
+                reinit="finish_previous",
             )
         else:
             wandb_logger = None