From cb10a25139cbe5baba3a34e8df0c277635b3277e Mon Sep 17 00:00:00 2001 From: ddilbaz Date: Thu, 30 Jan 2025 20:28:18 +0000 Subject: [PATCH] Make sure to follow latest format for tests and disable parse_shlo_mlir --- tests/models/mamba/test_mamba.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/models/mamba/test_mamba.py b/tests/models/mamba/test_mamba.py index 0aa473b2..4f1fe268 100644 --- a/tests/models/mamba/test_mamba.py +++ b/tests/models/mamba/test_mamba.py @@ -55,7 +55,6 @@ def set_model_eval(self, model): "state-spaces/mamba-370m-hf", ], ) - @pytest.mark.parametrize("op_by_op", [True, False], ids=["op_by_op", "full"]) def test_mamba(record_property, model_name, mode, op_by_op): record_property("model_name", model_name) @@ -70,7 +69,6 @@ def test_mamba(record_property, model_name, mode, op_by_op): ) results = tester.test_model() - if mode == "eval": gen_text = tester.tokenizer.batch_decode(results) print("Generated text: ", gen_text)