From 1ea24e8f90c1ceae169701a3e69cd1ff8af3f6a8 Mon Sep 17 00:00:00 2001
From: Benjamin Bossan <benjamin.bossan@gmail.com>
Date: Tue, 22 Apr 2025 11:21:03 +0200
Subject: [PATCH] TST AQLM test no longer x-fails

There has been an AQLM release with the fix to the issue that originally
broke the test.
---
 tests/test_gpu_examples.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/tests/test_gpu_examples.py b/tests/test_gpu_examples.py
index f591d52826..ff259bcdf4 100644
--- a/tests/test_gpu_examples.py
+++ b/tests/test_gpu_examples.py
@@ -2880,8 +2880,6 @@ def _check_inference_finite(self, model, batch):
         model.train(training)
 
     @pytest.mark.single_gpu_tests
-    # see https://github.com/Vahe1994/AQLM/pull/139
-    @pytest.mark.xfail(reason="AQLM does not work with PyTorch 2.5 (yet)", strict=True, raises=AttributeError)
     def test_causal_lm_training_aqlm(self):
         r"""
         Test the CausalLM training on a single GPU device. The test would simply fail if the adapters are not set