suppress old torch warning

ai4sd · Feb 3, 2025 · 98d1cbb · 98d1cbb
1 parent 974a741
commit 98d1cbb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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