Make sure to follow latest format for tests and disable parse_shlo_mlir

tenstorrent · Jan 30, 2025 · cb10a25 · cb10a25
1 parent 37ad3e7
commit cb10a25
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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)