fix hparams self tuning jax

Niccolo-Ajroldi · web-flow · commit 2db00ebbcee1 · 2025-03-14T10:10:17.000+01:00
diff --git a/prize_qualification_baselines/self_tuning/jax_nadamw_target_setting.py b/prize_qualification_baselines/self_tuning/jax_nadamw_target_setting.py
@@ -1,6 +1,5 @@
 """Submission file for an NAdamW optimizer with warmup+cosine LR in Jax."""
 
-import collections
 import functools
 
 # isort: off
