diff --git a/tests/e2e/trainer/run_trainer.py b/tests/e2e/trainer/run_trainer.py
index 1ba609b..40fc79d 100644
--- a/tests/e2e/trainer/run_trainer.py
+++ b/tests/e2e/trainer/run_trainer.py
@@ -1,7 +1,13 @@
 import math
 import os
+import warnings
 from pathlib import Path
 
+warnings.filterwarnings("ignore", module="mblm.model.mamba_shim", category=UserWarning)
+
+# https://github.com/pytorch/pytorch/issues/103444
+warnings.filterwarnings("ignore", module="torch.optim.lr_scheduler", category=UserWarning)
+
 import torch
 from torch.optim import Adam, Optimizer  # type: ignore
 from torch.optim.lr_scheduler import CosineAnnealingLR, LinearLR, LRScheduler, SequentialLR