diff --git a/tests/llmcompressor/transformers/obcq/test_consecutive_runs.py b/tests/llmcompressor/transformers/obcq/test_consecutive_runs.py index 686056099a..045c89a885 100644 --- a/tests/llmcompressor/transformers/obcq/test_consecutive_runs.py +++ b/tests/llmcompressor/transformers/obcq/test_consecutive_runs.py @@ -160,4 +160,4 @@ def setUp(self): self.output_second = Path(self.output) / "test_2" def test_consecutive_runs_gpu(self): - self._test_consecutive_runs(tolerance=1e-0, num_calibration_samples=1) + self._test_consecutive_runs(tolerance=1e-0)