diff --git a/tests/models/albert/test_albert_token_classification.py b/tests/models/albert/test_albert_token_classification.py index 9353216a..c17ee627 100644 --- a/tests/models/albert/test_albert_token_classification.py +++ b/tests/models/albert/test_albert_token_classification.py @@ -35,7 +35,6 @@ def _load_inputs(self): @pytest.mark.parametrize("model_name", ["albert/albert-base-v2"]) @pytest.mark.parametrize("op_by_op", [True, False], ids=["op_by_op", "full"]) def test_albert_token_classification(record_property, model_name, mode, op_by_op): - record_property("model_name", f"{model_name}-classification") cc = CompilerConfig() cc.enable_consteval = True diff --git a/tests/models/segformer/test_segformer.py b/tests/models/segformer/test_segformer.py index 0a1269d3..7a4e8ae0 100644 --- a/tests/models/segformer/test_segformer.py +++ b/tests/models/segformer/test_segformer.py @@ -68,4 +68,4 @@ def test_segformer(record_property, mode, op_by_op): if mode == "eval": logits = results.logits # shape (batch_size, num_labels, height/4, width/4) - results.finalize() + tester.finalize() diff --git a/tests/models/timm/test_timm_image_classification.py b/tests/models/timm/test_timm_image_classification.py index 06411204..29e469b5 100644 --- a/tests/models/timm/test_timm_image_classification.py +++ b/tests/models/timm/test_timm_image_classification.py @@ -64,9 +64,6 @@ def _load_inputs(self): @pytest.mark.parametrize("mode", ["train", "eval"]) @pytest.mark.parametrize("op_by_op", [True, False], ids=["op_by_op", "full"]) def test_timm_image_classification(record_property, model_name, mode, op_by_op): - record_property("model_name", model_name) - record_property("mode", mode) - if mode == "train": pytest.skip() cc = CompilerConfig()